Searched refs:PrepareToUpdateVariable (Results 1 – 4 of 4) sorted by relevance
199 Status PrepareToUpdateVariable(OpKernelContext* ctx, Tensor* tensor, in PrepareToUpdateVariable() function252 TF_RETURN_IF_ERROR(PrepareToUpdateVariable<Device, T>( in GetInputTensorFromVariable()
89 TF_RETURN_IF_ERROR(PrepareToUpdateVariable<Device, StateElementType>( in UpdateVariableAndFill()
512 context, PrepareToUpdateVariable<Device, T>( in Compute()
623 friend Status PrepareToUpdateVariable(