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

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 374 // Naming: diff_dst is input_gradient_tensor; src is orig_input_tensor.
523 Tensor input_gradient_tensor; local
534 input_gradient_tensor = ConvertMklToTF<T>(
537 input_gradient_tensor = MklGetInput(context, kIdxGradient);
554 const int64 batch = static_cast<int64>(input_gradient_tensor.dim_size(0));
555 const int64 rows = static_cast<int64>(input_gradient_tensor.dim_size(1));
556 const int64 cols = static_cast<int64>(input_gradient_tensor.dim_size(2));
557 const int64 depth = static_cast<int64>(input_gradient_tensor.dim_size(3));
561 input_gradient_tensor.shaped<T, 2>({nodes * batch, depth});
571 input_gradient_tensor.shape()
607 const Tensor& input_gradient_tensor = MklGetInput(context, kIdxGradient); local
    [all...]
mkl_pooling_ops_common.h 716 const Tensor& input_gradient_tensor,
725 ? TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(),
728 input_gradient_tensor.shape(),
733 &input_gradient_tensor);
745 : TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(),
mkl_avgpooling_op.cc 674 const Tensor& input_gradient_tensor,
694 context, input_gradient_tensor.dims() == 4,

Completed in 761 milliseconds