Searched defs:value_dtype (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/ops/ |
H A D | resource_variable_ops.cc | 38 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 D | lookup_util.cc | 76 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 D | lookup_ops.py | 78 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