| /external/tensorflow/tensorflow/python/ops/ |
| D | lookup_ops.py | 85 def _check_table_dtypes(table, key_dtype, value_dtype): argument
100 if value_dtype.base_dtype != table.value_dtype:
102 (table.value_dtype, value_dtype))
108 def __init__(self, key_dtype, value_dtype): argument
116 self._value_dtype = dtypes.as_dtype(value_dtype)
128 def value_dtype(self): member in LookupInterface
164 initializer.value_dtype)
283 value_dtype=self._initializer.value_dtype,
333 def __init__(self, key_dtype, value_dtype): argument
341 self._value_dtype = dtypes.as_dtype(value_dtype)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 76 DataType key_dtype, int64 key_index, DataType value_dtype, in Init() argument
82 value_ = Tensor(value_dtype, TensorShape({})); in Init()
327 DataType value_dtype, const string& table_name) { in CheckTableDataTypes() argument
328 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) { in CheckTableDataTypes()
331 DataTypeString(value_dtype), " with ", in CheckTableDataTypes()
333 DataTypeString(table.value_dtype()), " for table ", table_name); in CheckTableDataTypes()
349 const DataType& value_dtype = table->value_dtype(); in InitializeTableFromTextFile() local
356 if (value_index == kLineNumber && value_dtype != DT_INT64) { in InitializeTableFromTextFile()
359 DataTypeString(table->value_dtype())); in InitializeTableFromTextFile()
361 if (value_index == kWholeLine && value_dtype != DT_STRING) { in InitializeTableFromTextFile()
[all …]
|
| D | lookup_table_op.cc | 130 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
720 value_dtype(), TensorShape({num_buckets_, value_size}), &value_buckets_, in AllocateBuckets()
795 table->value_dtype()}; in Compute()
796 DataTypeVector expected_outputs = {table->value_dtype()}; in Compute()
831 table->value_dtype()}; in Compute()
941 table->value_dtype()}; in Compute()
966 #define REGISTER_KERNEL(key_dtype, value_dtype) \ argument
971 .TypeConstraint<value_dtype>("value_dtype"), \
[all …]
|
| D | ragged_gather_op_test.cc | 37 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); in BuildRaggedGatherGraph() local
44 .Input(FakeInput(value_dtype)) // params_dense_values in BuildRaggedGatherGraph()
48 .Attr("Tvalues", value_dtype) in BuildRaggedGatherGraph()
|
| D | lookup_table_op.h | 38 template <class Container, class key_dtype, class value_dtype>
83 DataTypeToEnum<value_dtype>::v(), cinfo_.name())); in Compute()
209 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); } in value_dtype() function
|
| D | lookup_util.h | 42 DataType value_dtype, const string& table_name);
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sharded_mutable_dense_hashtable.py | 60 value_dtype, argument
68 self._value_dtype = value_dtype
76 value_dtype=value_dtype,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | state_ops.cc | 286 DataType value_dtype; in __anone3ce2a1d0502() local
287 TF_RETURN_IF_ERROR(c->GetAttr("T", &value_dtype)); in __anone3ce2a1d0502()
288 if (value_dtype != shape_and_type.dtype) { in __anone3ce2a1d0502()
290 "Data types do not match: ", DataTypeString(value_dtype), " and ", in __anone3ce2a1d0502()
|
| D | resource_variable_ops.cc | 39 DataType value_dtype; in ValidateVariableResourceHandle() local
40 TF_RETURN_IF_ERROR(c->GetAttr("dtype", &value_dtype)); in ValidateVariableResourceHandle()
41 if (shape_and_type->at(0).dtype != value_dtype) { in ValidateVariableResourceHandle()
46 DataTypeString(value_dtype)); in ValidateVariableResourceHandle()
|
| D | lookup_ops.cc | 111 DataType value_dtype; in ValidateTableResourceHandle() local
112 TF_RETURN_IF_ERROR(c->GetAttr(value_dtype_attr, &value_dtype)); in ValidateTableResourceHandle()
113 if (value_shape_and_type.dtype != value_dtype) { in ValidateTableResourceHandle()
118 DataTypeString(value_dtype)); in ValidateTableResourceHandle()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.lookup.-key-value-tensor-initializer.pbtxt | 12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
|
| D | tensorflow.lookup.-text-file-initializer.pbtxt | 12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 21 name: "value_dtype"
26 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
|
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 21 name: "value_dtype"
|
| D | tensorflow.lookup.-static-hash-table.pbtxt | 26 name: "value_dtype"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lookup.-key-value-tensor-initializer.pbtxt | 12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'keys\', \'values\', \'key_dtype\', \'value_dtype\', \'name\'], varargs…
|
| D | tensorflow.lookup.-text-file-initializer.pbtxt | 12 name: "value_dtype"
17 …argspec: "args=[\'self\', \'filename\', \'key_dtype\', \'key_index\', \'value_dtype\', \'value_ind…
|
| D | tensorflow.lookup.experimental.-dense-hash-table.pbtxt | 21 name: "value_dtype"
26 …argspec: "args=[\'self\', \'key_dtype\', \'value_dtype\', \'default_value\', \'empty_key\', \'dele…
|
| D | tensorflow.lookup.-static-vocabulary-table.pbtxt | 26 name: "value_dtype"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | lookup_interface.cc | 39 if (values.dtype() != value_dtype()) { in CheckKeyAndValueTypes()
40 return errors::InvalidArgument("Value must be type ", value_dtype(), in CheckKeyAndValueTypes()
|
| D | lookup_interface.h | 98 virtual DataType value_dtype() const = 0;
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops.py | 333 value_dtype=self._initializer.value_dtype,
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | math_utils.py | 538 def __init__(self, key_dtype, value_dtype, default_value, *args, **kwargs): argument
542 value_dtype=value_dtype,
585 value_dtype=default_value.dtype.base_dtype,
613 def check_table_dtypes(self, key_dtype, value_dtype): argument
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MutableHashTableOfTensorsV2.pbtxt | 33 name: "value_dtype"
|
| D | api_def_MutableHashTableOfTensors.pbtxt | 31 name: "value_dtype"
|