Searched refs:PrepareToUpdateVariable (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 198 Status PrepareToUpdateVariable(OpKernelContext* ctx, Tensor* tensor, in PrepareToUpdateVariable() function
250 TF_RETURN_IF_ERROR(PrepareToUpdateVariable<Device, T>( in GetInputTensorFromVariable()
|
| D | stateful_random_ops.cc | 110 TF_RETURN_IF_ERROR(PrepareToUpdateVariable<Device, StateElementType>( in UpdateVariableAndFill()
283 OP_REQUIRES_OK(ctx, PrepareToUpdateVariable<Device, T>( in Compute()
|
| D | random_binomial_op.cc | 421 OP_REQUIRES_OK(ctx, PrepareToUpdateVariable<Device, StateElementType>( in Compute()
|
| D | resource_variable_ops.cc | 542 context, PrepareToUpdateVariable<Device, T>( in Compute()
|