Searched refs:var_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops.cc | 97 Tensor* var_tensor = var->tensor(); in UpdateVariableAndFill() local
98 TF_RETURN_IF_ERROR(CheckState(*var_tensor)); in UpdateVariableAndFill()
99 auto var_tensor_flat = var_tensor->flat<StateElementType>(); in UpdateVariableAndFill()
109 TF_RETURN_IF_ERROR(CheckPhiloxState(*var_tensor, alg_tag_skip)); in UpdateVariableAndFill()
111 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
117 arg.state_tensor = var_tensor; in UpdateVariableAndFill()
280 Tensor* var_tensor = var->tensor(); in Compute() local
281 OP_REQUIRES_OK(ctx, CheckState(*var_tensor)); in Compute()
284 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
297 const_cast<const Tensor*>(var_tensor)->flat<T>()); in Compute()
[all …]
|
| D | random_binomial_op.cc | 399 Tensor* var_tensor = var->tensor(); in Compute() local
401 ctx, var_tensor->dtype() == STATE_ELEMENT_DTYPE, in Compute()
404 DataTypeString(var_tensor->dtype()))); in Compute()
405 OP_REQUIRES(ctx, var_tensor->dims() == 1, in Compute()
408 var_tensor->dims())); in Compute()
409 auto var_tensor_flat = var_tensor->flat<StateElementType>(); in Compute()
422 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
|
| D | resource_variable_ops.cc | 555 Tensor* var_tensor = variable->tensor(); in Compute() local
556 OP_REQUIRES(context, var_tensor->shape().IsSameSize(value.shape()), in Compute()
558 var_tensor->shape().DebugString(), in Compute()
564 context, var_tensor, variable->copy_on_read_mode.load())); in Compute()
566 update_functor(context->eigen_device<Device>(), var_tensor->flat<T>(), in Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_experimental.cc | 216 Tensor* var_tensor = variable->tensor(); in TF_AssignUpdateVariable() local
218 context, var_tensor->shape().IsSameSize(value.shape()), in TF_AssignUpdateVariable()
220 var_tensor->shape().DebugString(), in TF_AssignUpdateVariable()
224 PrepareToUpdateVariable(ctx, var_tensor, in TF_AssignUpdateVariable()
228 TF_Tensor* tf_var_tensor = TF_TensorFromTensor(*var_tensor, &s); in TF_AssignUpdateVariable()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFixture.h | 60 void fill(U &&src_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U &&gamma_tensor) in fill() argument
68 library->fill(var_tensor, distribution_var, 0); in fill()
|
| D | BatchNormalizationLayerFusionFixture.h | 66 …void fill(U &&src, U &&w_tensor, U &&b_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U… in fill() argument
74 library->fill(var_tensor, distribution_gz, 3); in fill()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 5658 var_tensor = ged_ops.take_while_dataset(
5663 super(_TakeWhileDataset, self).__init__(input_dataset, var_tensor)
|