Home
last modified time | relevance | path

Searched defs:orig_input_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc46 const Tensor& orig_input_tensor, in SpaceToBatchOpCompute()
226 const Tensor& orig_input_tensor = context->input(0); in Compute() local
H A Dbatchtospace_op.cc44 const Tensor& orig_input_tensor, in BatchToSpaceOpCompute()
210 const Tensor& orig_input_tensor = context->input(0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc373 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in Compute() local
505 Tensor orig_input_tensor; in MklDefaultToEigen() local
593 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in SanityCheckInputs() local
H A Dmkl_avgpooling_op.cc189 const Tensor& orig_input_tensor = in Compute() local
H A Dmkl_maxpooling_op.cc243 const Tensor& orig_input_tensor = in Compute() local