| /external/tensorflow/tensorflow/python/ops/ |
| D | map_ops.py | 40 def tensor_map_lookup(input_handle, key, value_dtype): argument
43 def tensor_map_erase(input_handle, key, value_dtype): argument
|
| D | lookup_ops.py | 111 def check_table_dtypes(table, key_dtype, value_dtype): argument
134 def __init__(self, key_dtype, value_dtype): argument
154 def value_dtype(self): member in LookupInterface
424 def __init__(self, key_dtype, value_dtype): argument
440 def value_dtype(self): member in TableInitializerBase
476 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
609 value_dtype, argument
1752 value_dtype, argument
2027 value_dtype, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_hashtables.cc | 59 auto value_dtype = hashtable_op.value_dtype(); in matchAndRewrite() local
136 auto value_dtype = hashtable.value_dtype(); in checkWhetherGraphHasValidStaticLookupTables() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | ragged_tensor_from_variant_op.cc | 31 const Tensor& encoded_variant, int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant()
248 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() 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
1115 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1149 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1182 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
|
| D | ragged_gather_op_test.cc | 39 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedGatherGraph() local
|
| D | lookup_table_op.h | 254 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function
|
| D | ragged_tensor_to_variant_op.cc | 220 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
|
| D | ragged_tensor_to_tensor_op_test.cc | 58 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedTensorToTensorGraph() 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 __anonc41c59800502() local
|
| D | lookup_ops.cc | 104 DataType value_dtype; in ValidateTableType() local
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | static_hashtable.cc | 98 TfLiteType value_dtype) { in CreateHashtableResourceIfNotAvailable()
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | index_lookup.py | 61 def __init__(self, key_dtype, value_dtype): argument
77 def value_dtype(self): member in _NullInitializer
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_test.py | 94 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_ops_test.cc | 629 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 | 2315 DataType value_dtype; in ValidateVariableResourceHandle() local
|