Searched refs:GetInputTensorFromVariable (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops.cc | 972 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in Compute()
1066 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
1069 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoValidate()
1073 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoValidate()
1122 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1125 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1129 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoCompute()
1205 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1208 OP_REQUIRES_OK(ctx, GetInputTensorFromVariable<Device, T>( in DoCompute()
1212 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoCompute()
[all …]
|
| H A D | training_op_helpers.h | 236 Status GetInputTensorFromVariable(OpKernelContext* ctx, int input, in GetInputTensorFromVariable() function
|