Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc44 const Tensor& orig_input_tensor, in BatchToSpaceOpCompute()
210 const Tensor& orig_input_tensor = context->input(0); in Compute() local
Dspacetobatch_op.cc46 const Tensor& orig_input_tensor, in SpaceToBatchOpCompute()
226 const Tensor& orig_input_tensor = context->input(0); in Compute() local
/external/tensorflow/tensorflow/core/kernels/mkl/
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
Dmkl_avgpooling_op.cc189 const Tensor& orig_input_tensor = in Compute() local
Dmkl_maxpooling_op.cc243 const Tensor& orig_input_tensor = in Compute() local