Searched defs:value_dtype (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
H A Dresource_variable_ops.cc38 DataType value_dtype; local
39 TF_RETURN_IF_ERROR(c->GetAttr("dtype", &value_dtype));
40 if (shape_and_type->dtype != value_dtype) {
45 DataTypeString(value_dtype));
/external/tensorflow/tensorflow/core/kernels/
H A Dlookup_util.cc76 DataType key_dtype, int64 key_index, DataType value_dtype,
85 value_ = Tensor(value_dtype, TensorShape({}));
315 DataType value_dtype, const string& table_name) {
316 if (table.key_dtype() != key_dtype || table.value_dtype() != value_dtype) {
318 "Conflicting key/value dtypes ", key_dtype, "->", value_dtype, " with ",
319 table.key_dtype(), "-", table.value_dtype(), " for table ", table_name);
335 const DataType& value_dtype = table->value_dtype(); local
342 if (value_index == kLineNumber && value_dtype !
75 Init(const string& filename, int64 vocab_size, char delimiter, DataType key_dtype, int64 key_index, DataType value_dtype, int64 value_index, Env* env) argument
314 CheckTableDataTypes(const LookupInterface& table, DataType key_dtype, DataType value_dtype, const string& table_name) argument
[all...]
/external/tensorflow/tensorflow/python/ops/
H A Dlookup_ops.py78 def _check_table_dtypes(table, key_dtype, value_dtype):
79 """Check that the given key_dtype and value_dtype matches the table dtypes.
84 value_dtype: The value data type to check.
87 TypeError: when 'key_dtype' or 'value_dtype' doesn't match the table data
93 if value_dtype.base_dtype != table.value_dtype:
95 (table.value_dtype, value_dtype))
101 def __init__(self, key_dtype, value_dtype, name):
106 value_dtype
119 def value_dtype(self): member in class:LookupInterface
307 def value_dtype(self): member in class:TableInitializerBase
[all...]

Completed in 228 milliseconds