| /external/tensorflow/tensorflow/core/kernels/ |
| lookup_util.h | 39 // Verify that the given key_dtype and value_dtype matches the corresponding
42 DataType value_dtype, const string& table_name);
|
| lookup_util.cc | 76 DataType key_dtype, int64 key_index, DataType value_dtype,
82 value_ = Tensor(value_dtype, TensorShape({}));
327 DataType value_dtype, const string& table_name) {
328 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) {
331 DataTypeString(value_dtype), " with ",
333 DataTypeString(table.value_dtype()), " for table ", table_name);
349 const DataType& value_dtype = table->value_dtype(); local
356 if (value_index == kLineNumber && value_dtype != DT_INT64)
[all...] |
| lookup_table_op.cc | 130 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); }
268 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); }
554 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); }
720 value_dtype(), TensorShape({num_buckets_, value_size}), &value_buckets_,
795 table->value_dtype()};
796 DataTypeVector expected_outputs = {table->value_dtype()};
831 table->value_dtype()};
[all...] |
| lookup_table_op.h | 36 // key and value are of the templated type "key_dtype" and "value_dtype"
38 template <class Container, class key_dtype, class value_dtype>
83 DataTypeToEnum<value_dtype>::v(), cinfo_.name()));
209 DataType value_dtype() const override { return DataTypeToEnum<V>::v(); }
|
| ragged_gather_op_test.cc | 37 const auto& value_dtype = DataTypeToEnum<VALUE_TYPE>::v(); local
44 .Input(FakeInput(value_dtype)) // params_dense_values
48 .Attr("Tvalues", value_dtype)
|
| lookup_table_init_op.cc | 56 table->value_dtype()};
|
| /external/tensorflow/tensorflow/core/framework/ |
| lookup_interface.cc | 39 if (values.dtype() != value_dtype()) {
40 return errors::InvalidArgument("Value must be type ", value_dtype(),
|
| lookup_interface.h | 98 virtual DataType value_dtype() const = 0;
110 // - DataType of the tensor values equals to the table value_dtype
130 // - DataType of the tensor default_value equals to the table value_dtype
148 // key_dtype and value_dtype.
|
| /external/tensorflow/tensorflow/python/ops/ |
| lookup_ops.py | 85 def _check_table_dtypes(table, key_dtype, value_dtype):
86 """Check that the given key_dtype and value_dtype matches the table dtypes.
91 value_dtype: The value data type to check.
94 TypeError: when 'key_dtype' or 'value_dtype' doesn't match the table data
100 if value_dtype.base_dtype != table.value_dtype:
102 (table.value_dtype, value_dtype))
108 def __init__(self, key_dtype, value_dtype):
113 value_dtype: The table value type
128 def value_dtype(self): member in class:LookupInterface
349 def value_dtype(self): member in class:TableInitializerBase
[all...] |
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| sharded_mutable_dense_hashtable.py | 60 value_dtype,
68 self._value_dtype = value_dtype
76 value_dtype=value_dtype,
|
| sdca_ops.py | 163 value_dtype=dtypes.float32,
|
| /external/tensorflow/tensorflow/core/ops/ |
| state_ops.cc | 286 DataType value_dtype;
287 TF_RETURN_IF_ERROR(c->GetAttr("T", &value_dtype));
288 if (value_dtype != shape_and_type.dtype) {
290 "Data types do not match: ", DataTypeString(value_dtype), " and ",
|
| lookup_ops.cc | 111 DataType value_dtype; local
112 TF_RETURN_IF_ERROR(c->GetAttr(value_dtype_attr, &value_dtype));
113 if (value_shape_and_type.dtype != value_dtype) {
118 DataTypeString(value_dtype));
327 TF_RETURN_IF_ERROR(c->GetAttr("value_dtype", &value_t));
342 .Attr("value_dtype: type")
352 .Attr("value_dtype: type")
362 .Attr("value_dtype: type")
372 .Attr("value_dtype: type")
385 .Attr("value_dtype: type"
[all...] |
| resource_variable_ops.cc | 39 DataType value_dtype; local
40 TF_RETURN_IF_ERROR(c->GetAttr("dtype", &value_dtype));
41 if (shape_and_type->at(0).dtype != value_dtype) {
46 DataTypeString(value_dtype));
|
| /external/tensorflow/tensorflow/python/training/ |
| saver_test_utils.py | 39 key_dtype=dtypes.string, value_dtype=dtypes.float32, name=name)
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| lookup_ops.py | 333 value_dtype=self._initializer.value_dtype,
|
| lookup_ops_test.py | 90 keys, values, value_dtype=dtypes.int64),
220 ["a"], [1], value_dtype=dtypes.int64),
[all...] |
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| math_utils.py | 538 def __init__(self, key_dtype, value_dtype, default_value, *args, **kwargs):
542 value_dtype=value_dtype,
585 value_dtype=default_value.dtype.base_dtype,
613 def check_table_dtypes(self, key_dtype, value_dtype):
[all...] |
| estimators.py | 271 for feature_key, (batch_only_feature_shape, value_dtype) in (
278 dtype=value_dtype, name=feature_key, shape=feature_shape)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_tensor_dense_matmul_op_test.py | 92 def _testBasic(self, value_dtype, indices_dtype=np.int64):
93 x = _maybe_complex(np.random.rand(10, 10).astype(value_dtype))
96 y = _maybe_complex(np.random.randn(10, 20).astype(value_dtype))
|
| lookup_ops_test.py | 113 keys, values, value_dtype=dtypes.int64), default_val)
239 value_dtype=dtypes.int64),
[all...] |
| /external/tensorflow/tensorflow/contrib/text/python/ops/ |
| skip_gram_ops.py | 365 value_dtype=vocab_freq_dtype,
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| load_v1_in_v2_test.py | 164 value_dtype=dtypes.int64,
|
| save_test.py | 341 value_dtype=dtypes.int64,
|
| /external/tensorflow/tensorflow/python/framework/ |
| ops.py | [all...] |
