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_pooling_ops_common.h 365 const Tensor& input_gradient_tensor,
373 TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(),
378 &input_gradient_tensor);
390 : TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(),
mkl_lrn_op.cc     [all...]
mkl_avgpooling_op.cc 554 const Tensor& input_gradient_tensor = variable
561 SanityCheckInputs(context, tensor_in_shape, input_gradient_tensor,
588 input_gradient_mkl_shape, input_gradient_tensor,
672 const Tensor& input_gradient_tensor,
691 OP_REQUIRES(context, input_gradient_tensor.dims() == 4,

Completed in 143 milliseconds