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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 195 const MklDnnShape& input_mkl_shape,
203 input_mkl_shape.IsMklTensor()
204 ? input_mkl_shape.GetMklLayout()
209 this->InitMklPoolParameters(context, pool_params, input_mkl_shape,
254 const MklDnnShape& input_mkl_shape) {
255 if (!input_mkl_shape.IsMklTensor()) {
259 OP_REQUIRES(context, input_mkl_shape.GetDimension() == 4,
mkl_conv_ops.h 368 MklDnnShape input_mkl_shape, filter_mkl_shape, outbprop_mkl_shape; variable
369 GetMklShape(context, kInputIdx, &input_mkl_shape);
373 ValidateMklShapes(input_mkl_shape, filter_mkl_shape, outbprop_mkl_shape);
429 input_mkl_shape.IsMklTensor()
430 ? input_mkl_shape.GetMklLayout()
503 virtual void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
mkl_conv_grad_input_ops.cc 372 void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
378 CHECK(!input_mkl_shape.IsMklTensor())
mkl_conv_grad_filter_ops.cc 447 void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 791 MklDnnShape input_mkl_shape; local
792 GetMklShape(context, input_idx, &input_mkl_shape);
793 if (input_mkl_shape.IsMklTensor()) {
794 return input_mkl_shape.GetTfShape();
    [all...]

Completed in 704 milliseconds