Searched defs:tensor_dict (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | parsing_config.py | 668 def _construct_tensors_for_composite_features(features, tensor_dict): argument
759 def _add_ragged_partition(values, partition, tensor_dict, row_splits_dtype, argument
804 def _add_batched_ragged_partition(rt, partition, tensor_dict, feature_key, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | node.py | 143 def map_arguments(self, tensor_dict): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | save.py | 472 def _tensor_dict_to_tensorinfo(tensor_dict): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 376 auto tensor_dict = get_tensor_dict(); in initialize_python_bindings() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 1184 c10::Dict<int64_t, Tensor> tensor_dict; in TEST() local
|