keras/keras_core/operations
Tirth Patel 1f1ea1436c Add dtype to keras_core.operations.one_hot (#327)
* Add dtype to keras_core.operations.one_hot

* Default dtype to None and use `backend.floatx()` instead

* Use `backend.floatx()` in the __init__ itself
2023-06-12 19:10:23 +00:00
..
__init__.py Add core ops shape (#306) 2023-06-08 17:13:29 -07:00
core_test.py Add core ops shape (#306) 2023-06-08 17:13:29 -07:00
core.py Add core ops shape (#306) 2023-06-08 17:13:29 -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 Add the stop_gradient function to keras_core.backend (#282) 2023-06-08 04:30:26 +00:00
math_test.py Add QR op and fix orthogonal initializer. 2023-06-10 11:58:13 -07:00
math.py Add QR op and fix orthogonal initializer. 2023-06-10 11:58:13 -07:00
nn_test.py Add dtype to keras_core.operations.one_hot (#327) 2023-06-12 19:10:23 +00:00
nn.py Add dtype to keras_core.operations.one_hot (#327) 2023-06-12 19:10:23 +00: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 Small ops fixes for Torch unit tests (#316) 2023-06-12 16:20:31 +00: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 Fix issue where looking up a tf.Tensor in the dictionary within SymbolicArguments would fail. (#288) 2023-06-07 12:13:22 -07:00