keras/keras_core/operations
Haifeng Jin 5b1470a35e fix math.segment_sum for torch backend (#263)
Co-authored-by: Haifeng Jin <haifeng-jin@users.noreply.github.com>
2023-06-05 09:14:46 -07:00
..
__init__.py Add IoU metrics: IoU, BinaryIoU, OneHotIoU, OneHotMeanIoU, (#127) 2023-05-15 20:50:55 -04:00
core_test.py Merge branch 'main' of github.com:keras-team/keras-core 2023-05-30 15:55:24 -07:00
core.py Enable a couple of KPL in tf.data 2023-06-04 20:55:24 -07:00
function_test.py Allow dynamic shapes for JAX. 2023-05-20 19:01:01 -07:00
function.py Add docstrings to major classes 2023-05-22 12:01:29 -07:00
image_test.py Adds torch image resizing and torchvision dependency. (#220) 2023-05-30 16:19:01 +00:00
image.py Enable jit compile in training tests; fix some bugs. 2023-05-19 13:40:22 -07:00
math_test.py fix math.segment_sum for torch backend (#263) 2023-06-05 09:14:46 -07:00
math.py Add torch.math.* (#189) 2023-05-18 18:04:25 -07:00
nn_test.py Minor refactor 2023-05-23 16:01:22 -07:00
nn.py Merge branch 'main' of github.com:keras-team/keras-core 2023-05-19 11:40:25 -07:00
node.py Format code 2023-04-26 11:59:47 -07:00
numpy_test.py Merge branch 'main' of github.com:keras-team/keras-core 2023-06-01 18:49:07 -07:00
numpy.py Merge branch 'main' of github.com:keras-team/keras-core 2023-06-01 18:49:07 -07:00
operation_test.py Refactor backend. 2023-05-03 15:33:40 -07:00
operation_utils_test.py Add get_source_inputs and tests 2023-05-22 12:01:53 -07:00
operation_utils.py Merge branch 'main' of github.com:keras-team/keras-core 2023-05-22 12:02:11 -07:00
operation.py Refactor random namespace 2023-05-15 15:01:46 -07:00
symbolic_arguments.py Merge branch 'main' of github.com:keras-team/keras-core 2023-06-01 18:49:07 -07:00