Searched refs:GetInputTensorFromVariable (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.cc | 472 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in Compute()
514 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<SYCLDevice, T>( in Compute()
631 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
634 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
638 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoValidate()
687 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
690 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
694 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoCompute()
779 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<CPUDevice, T>( in DoCompute()
782 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<CPUDevice, T>( in DoCompute()
[all …]
|
| D | training_op_helpers.h | 241 Status GetInputTensorFromVariable(OpKernelContext* ctx, int input, in GetInputTensorFromVariable() function
|