| /external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/ |
| D | simple_hash_table_test.py | 32 def _use_table(self, key_dtype, value_dtype): argument
45 def test_find_insert_find_eager(self, key_dtype, value_dtype): argument
54 def test_find_insert_find_tf_function(self, key_dtype, value_dtype): argument
|
| D | simple_hash_table.py | 53 value_dtype, argument
116 def value_dtype(self): member in SimpleHashTable
|
| D | simple_hash_table_op.cc | 79 DataType value_dtype; in ValidateTableType() local
|
| D | simple_hash_table_kernel.cc | 279 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | map_ops.py | 36 def tensor_map_lookup(input_handle, key, value_dtype): argument
39 def tensor_map_erase(input_handle, key, value_dtype): argument
|
| D | lookup_ops.py | 108 def check_table_dtypes(table, key_dtype, value_dtype): argument
131 def __init__(self, key_dtype, value_dtype): argument
151 def value_dtype(self): member in LookupInterface
479 def __init__(self, key_dtype, value_dtype): argument
495 def value_dtype(self): member in TableInitializerBase
531 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
664 value_dtype, argument
1819 value_dtype, argument
2120 value_dtype, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_hashtables.cc | 64 auto value_dtype = hashtable_op.value_dtype(); in matchAndRewrite() local
141 auto value_dtype = hashtable.value_dtype(); in checkWhetherGraphHasValidStaticLookupTables() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_ops_test.cc | 55 typedef int32 value_dtype; typedef
|
| D | lookup_util.cc | 84 DataType key_dtype, int64_t key_index, DataType value_dtype, in Init()
350 DataType value_dtype, const string& table_name) { in CheckTableDataTypes()
382 const DataType& value_dtype = table->value_dtype(); in InitializeTableFromTextFile() local
|
| D | lookup_table_op.cc | 140 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
328 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
663 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableDenseHashTable
1080 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1122 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1164 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1205 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| D | ragged_tensor_from_variant_op.cc | 34 int output_ragged_rank, DataType value_dtype, DataType split_dtype, in RaggedComponentsFromVariant()
310 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
|
| D | ragged_gather_op_test.cc | 39 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedGatherGraph() local
|
| D | lookup_table_op.h | 288 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function
|
| D | ragged_tensor_to_tensor_op_test.cc | 58 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedTensorToTensorGraph() local
|
| D | ragged_tensor_to_variant_op.cc | 270 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
|
| D | ragged_tensor_from_variant_op_test.cc | 45 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildDecodeRaggedTensorGraph() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | state_ops.cc | 340 DataType value_dtype; in __anon70548b590502() local
|
| D | lookup_ops.cc | 104 DataType value_dtype; in ValidateTableType() local
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 100 TfLiteType value_dtype) { in CreateHashtableResourceIfNotAvailable()
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_tensor_dense_matmul_op_test.py | 90 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_ops_test.cc | 631 TensorType value_dtype) { in HashtableOpModel()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 497 TfLiteType value_dtype; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2366 DataType value_dtype; in ValidateVariableResourceHandle() local
|