Home
last modified time | relevance | path

Searched defs:value_dtype (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dmap_ops.py40 def tensor_map_lookup(input_handle, key, value_dtype): argument
43 def tensor_map_erase(input_handle, key, value_dtype): argument
Dlookup_ops.py89 def _check_table_dtypes(table, key_dtype, value_dtype): argument
112 def __init__(self, key_dtype, value_dtype): argument
132 def value_dtype(self): member in LookupInterface
402 def __init__(self, key_dtype, value_dtype): argument
418 def value_dtype(self): member in TableInitializerBase
513 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
646 value_dtype, argument
1743 value_dtype, argument
2013 value_dtype, argument
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable.cc41 TfLiteType value_dtype; member
51 TfLiteType key_dtype, value_dtype; in InitHashtable() local
Dhashtable_ops_test.cc638 TensorType value_dtype) { in HashtableOpModel()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_from_variant_op.cc31 const Tensor& encoded_variant, int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant()
232 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
Dlookup_table_op.cc141 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
290 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
576 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableDenseHashTable
1000 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1035 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1069 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1102 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
Dlookup_util.cc79 DataType key_dtype, int64 key_index, DataType value_dtype, in Init()
344 DataType value_dtype, const string& table_name) { in CheckTableDataTypes()
366 const DataType& value_dtype = table->value_dtype(); in InitializeTableFromTextFile() local
Dragged_gather_op_test.cc39 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedGatherGraph() local
Dlookup_table_op.h217 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function
Dragged_tensor_to_variant_op.cc209 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
Dragged_tensor_to_tensor_op_test.cc58 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedTensorToTensorGraph() local
Dragged_tensor_from_variant_op_test.cc45 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildDecodeRaggedTensorGraph() local
/external/tensorflow/tensorflow/core/ops/
Dstate_ops.cc340 DataType value_dtype; in __anonaa115abb0502() local
Dlookup_ops.cc104 DataType value_dtype; in ValidateTableType() local
/external/tensorflow/tensorflow/lite/experimental/resource/
Dstatic_hashtable.cc98 TfLiteType value_dtype) { in CreateHashtableResourceIfNotAvailable()
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_tensor_dense_matmul_op_test.py92 def _testBasic(self, value_dtype, indices_dtype=np.int64): argument
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc2275 DataType value_dtype; in ValidateVariableResourceHandle() local