HomeSort by relevance Sort by last modified time
    Searched refs:orig_output_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 525 Tensor orig_output_tensor; local
548 orig_output_tensor = ConvertMklToTF<T>(
551 orig_output_tensor = MklGetInput(context, kIdxOrigOutput);
564 auto activations = orig_output_tensor.shaped<T, 2>({nodes * batch, depth});
609 const Tensor& orig_output_tensor = MklGetInput(context, kIdxOrigOutput); local
643 context, orig_output_tensor.dims() == 4,
mkl_maxpooling_op.cc     [all...]

Completed in 1018 milliseconds