Home
last modified time | relevance | path

Searched refs:value_dtype (Results 1 – 25 of 90) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable.cc41 TfLiteType value_dtype; member
51 TfLiteType key_dtype, value_dtype; in InitHashtable() local
55 &value_dtype, nullptr); in InitHashtable()
60 option->value_dtype = value_dtype; in InitHashtable()
79 params->value_dtype == kTfLiteString) || in PrepareHashtable()
81 params->value_dtype == kTfLiteInt64)); in PrepareHashtable()
121 &resources, resource_id, params->key_dtype, params->value_dtype); in EvalHashtable()
/external/tensorflow/tensorflow/python/ops/
Dlookup_ops.py89 def _check_table_dtypes(table, key_dtype, value_dtype): argument
104 if value_dtype.base_dtype != table.value_dtype:
106 (table.value_dtype, value_dtype))
112 def __init__(self, key_dtype, value_dtype): argument
120 self._value_dtype = dtypes.as_dtype(value_dtype)
132 def value_dtype(self): member in LookupInterface
172 initializer.value_dtype)
321 value_dtype=self._initializer.value_dtype,
402 def __init__(self, key_dtype, value_dtype): argument
410 self._value_dtype = dtypes.as_dtype(value_dtype)
[all …]
Dmap_ops.py40 def tensor_map_lookup(input_handle, key, value_dtype): argument
41 return gen_map_ops.tensor_map_lookup(input_handle, key, value_dtype)
43 def tensor_map_erase(input_handle, key, value_dtype): argument
44 return gen_map_ops.tensor_map_erase(input_handle, key, value_dtype)
/external/tensorflow/tensorflow/core/kernels/
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
742 value_dtype(), TensorShape({num_buckets_, value_size}), &value_buckets_, in AllocateBuckets()
835 table->value_dtype()}; in Compute()
836 DataTypeVector expected_outputs = {table->value_dtype()}; in Compute()
869 table->value_dtype()}; in Compute()
975 table->value_dtype()}; in Compute()
1000 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
1005 .TypeConstraint<value_dtype>("value_dtype"), \
[all …]
Dlookup_util.cc79 DataType key_dtype, int64 key_index, DataType value_dtype, in Init() argument
85 value_ = Tensor(value_dtype, TensorShape({})); in Init()
344 DataType value_dtype, const string& table_name) { in CheckTableDataTypes() argument
345 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) { in CheckTableDataTypes()
348 DataTypeString(value_dtype), " with ", in CheckTableDataTypes()
350 DataTypeString(table.value_dtype()), " for table ", table_name); in CheckTableDataTypes()
366 const DataType& value_dtype = table->value_dtype(); in InitializeTableFromTextFile() local
373 if (value_index == kLineNumber && value_dtype != DT_INT64) { in InitializeTableFromTextFile()
376 DataTypeString(table->value_dtype())); in InitializeTableFromTextFile()
378 if (value_index == kWholeLine && value_dtype != DT_STRING) { in InitializeTableFromTextFile()
[all …]
Dragged_tensor_from_variant_op.cc31 const Tensor& encoded_variant, int ragged_rank, DataType value_dtype, in RaggedComponentsFromVariant() argument
53 if (decoded->values().dtype() != value_dtype) { in RaggedComponentsFromVariant()
55 "Expected values Tensor dtype: ", DataTypeString(value_dtype), in RaggedComponentsFromVariant()
232 const auto value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in Compute() local
237 value_dtype, split_dtype, &decoded_components)); in Compute()
Dragged_gather_op_test.cc39 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedGatherGraph() local
46 .Input(FakeInput(value_dtype)) // params_dense_values in BuildRaggedGatherGraph()
50 .Attr("Tvalues", value_dtype) in BuildRaggedGatherGraph()
Dragged_tensor_to_tensor_op_test.cc58 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedTensorToTensorGraph() local
63 .Attr("T", value_dtype) in BuildRaggedTensorToTensorGraph()
68 .Input(FakeInput(value_dtype)) // values in BuildRaggedTensorToTensorGraph()
69 .Input(FakeInput(value_dtype)) // default_value in BuildRaggedTensorToTensorGraph()
Dlookup_table_op.h39 template <class Container, class key_dtype, class value_dtype>
92 DataTypeToEnum<value_dtype>::v(), cinfo_.name())); in Compute()
217 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorMapLookup.pbtxt13 type_attr: "value_dtype"
20 name: "value_dtype"
DTensorMapInsert.pbtxt13 type_attr: "value_dtype"
24 name: "value_dtype"
DTensorMapErase.pbtxt20 name: "value_dtype"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dinit_text_file_to_import_invalid.mlir7 …me = "hash_table_/tmp/vocab.txt_-2_-1", use_node_name_sharing = false, value_dtype = i64} : () -> …
19 …me = "hash_table_/tmp/vocab.txt_-2_-1", use_node_name_sharing = false, value_dtype = i64} : () -> …
33 …me = "hash_table_/tmp/vocab.txt_-2_-1", use_node_name_sharing = false, value_dtype = i64} : () -> …
Dinit_text_file_to_import.mlir7 …me = "hash_table_/tmp/vocab.txt_-2_-1", use_node_name_sharing = false, value_dtype = i64} : () -> …
21 …me = "hash_table_/tmp/vocab.txt_-2_-1", use_node_name_sharing = false, value_dtype = i64} : () -> …
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
Dtensorflow.lookup.-text-file-initializer.pbtxt12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
Dtensorflow.lookup.experimental.-dense-hash-table.pbtxt22 name: "value_dtype"
27 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lookup.-key-value-tensor-initializer.pbtxt12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
Dtensorflow.lookup.-text-file-initializer.pbtxt12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
Dtensorflow.lookup.experimental.-dense-hash-table.pbtxt22 name: "value_dtype"
27 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
/external/tensorflow/tensorflow/core/ops/
Dstate_ops.cc340 DataType value_dtype; in __anonaa115abb0502() local
341 TF_RETURN_IF_ERROR(c->GetAttr("T", &value_dtype)); in __anonaa115abb0502()
342 if (value_dtype != shape_and_type.dtype) { in __anonaa115abb0502()
344 "Data types do not match: ", DataTypeString(value_dtype), " and ", in __anonaa115abb0502()
/external/tensorflow/tensorflow/lite/experimental/resource/
Dstatic_hashtable.cc98 TfLiteType value_dtype) { in CreateHashtableResourceIfNotAvailable() argument
102 auto* hashtable = internal::CreateStaticHashtable(key_dtype, value_dtype); in CreateHashtableResourceIfNotAvailable()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dforce_shared_name_for_resource_ops.pbtxt19 key: "value_dtype"
123 key: "value_dtype"
/external/tensorflow/tensorflow/core/framework/
Dlookup_interface.cc39 if (values.dtype() != value_dtype()) { in CheckKeyAndValueTypes()
40 return errors::InvalidArgument("Value must be type ", value_dtype(), in CheckKeyAndValueTypes()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dinsert_call_once_op.mlir12 …_dba2ccaa-f1b1-46d6-b276-98008f69da71", use_node_name_sharing = false, value_dtype = !tf.string} :…
21 …_dba2ccaa-f1b1-46d6-b276-98008f69da71", use_node_name_sharing = false, value_dtype = !tf.string} :…

1234