Searched refs:var_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops.cc | 56 Tensor* var_tensor = var->tensor(); in UpdateVariableAndFill() local
57 if (var_tensor->dtype() != STATE_ELEMENT_DTYPE) { in UpdateVariableAndFill()
61 DataTypeString(var_tensor->dtype())); in UpdateVariableAndFill()
63 if (var_tensor->dims() != 1) { in UpdateVariableAndFill()
66 var_tensor->dims()); in UpdateVariableAndFill()
68 auto var_tensor_flat = var_tensor->flat<StateElementType>(); in UpdateVariableAndFill()
90 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
93 &state_var_guard, var_tensor, output_data); in UpdateVariableAndFill()
|
| D | resource_variable_ops.cc | 504 Tensor* var_tensor = variable->tensor(); in Compute() local
505 OP_REQUIRES(context, var_tensor->shape().IsSameSize(value.shape()), in Compute()
507 var_tensor->shape().DebugString(), in Compute()
513 context, var_tensor, variable->copy_on_read_mode.load())); in Compute()
515 update_functor(context->eigen_device<Device>(), var_tensor->flat<T>(), in Compute()
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| D | zero_initializer_op.cc | 104 Tensor* var_tensor = nullptr; in Compute() local
109 dtype_, shape_, &unused, &var_tensor, attr)); in Compute()
113 var_tensor->flat<T>()); in Compute()
115 *(*var_ptr)->tensor() = *var_tensor; in Compute()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | take_while_ops.py | 44 var_tensor = gen_experimental_dataset_ops.experimental_take_while_dataset(
49 super(_TakeWhileDataset, self).__init__(input_dataset, var_tensor)
|