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()
214 const Tensor& orig_input_tensor = context->input(0); in Compute() local
Dmkl_lrn_op.cc376 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in Compute() local
524 Tensor orig_input_tensor; in MklDefaultToEigen() local
608 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in SanityCheckInputs() local
Dmkl_maxpooling_op.cc689 const Tensor& orig_input_tensor = in Compute() local
813 const Tensor& orig_input_tensor, in SanityCheckInputs()
Dmkl_avgpooling_op.cc576 const Tensor& orig_input_tensor = in Compute() local