/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...] |