Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/experimental/kernels/
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/python/ops/
Dlookup_ops.py87 def _check_table_dtypes(table, key_dtype, value_dtype): argument
110 def __init__(self, key_dtype, value_dtype): argument
130 def value_dtype(self): member in LookupInterface
378 def __init__(self, key_dtype, value_dtype): argument
394 def value_dtype(self): member in TableInitializerBase
418 def __init__(self, keys, values, key_dtype=None, value_dtype=None, name=None): argument
550 value_dtype, argument
1609 value_dtype, argument
1858 value_dtype, argument
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc76 DataType key_dtype, int64 key_index, DataType value_dtype, in Init()
337 DataType value_dtype, const string& table_name) { in CheckTableDataTypes()
359 const DataType& value_dtype = table->value_dtype(); in InitializeTableFromTextFile() local
Dlookup_table_op.cc130 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfScalars
268 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableHashTableOfTensors
554 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function in tensorflow::lookup::MutableDenseHashTable
978 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1013 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1047 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1080 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
Dragged_tensor_from_variant_op.cc35 int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant()
252 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() 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_tensor_op_test.cc58 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedTensorToTensorGraph() local
Dragged_tensor_from_variant_op_test.cc44 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildDecodeRaggedTensorGraph() local
/external/tensorflow/tensorflow/core/ops/
Dstate_ops.cc286 DataType value_dtype; in __anon5d8c585f0502() local
Dlookup_ops.cc111 DataType value_dtype; in ValidateTableResourceHandle() 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.cc2169 DataType value_dtype; in ValidateVariableResourceHandle() local