Searched refs:GetInputTensorFromVariable (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.cc | 944 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in Compute()
1042 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
1045 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
1049 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoValidate()
1098 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1101 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1105 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoCompute()
1186 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<CPUDevice, T>( in DoCompute()
1189 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<CPUDevice, T>( in DoCompute()
1193 GetInputTensorFromVariable<CPUDevice, T>( in DoCompute()
[all …]
|
| D | training_op_helpers.h | 240 Status GetInputTensorFromVariable(OpKernelContext* ctx, int input, in GetInputTensorFromVariable() function
|