Searched refs:output_dims_tf_order (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 321 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
324 DCHECK(output_dims_tf_order);
460 *output_dims_tf_order = TFShapeToMklDnnDims(out_shape);
487 const memory::dims& dilations, memory::dims* output_dims_tf_order, in GetOutputAndPadSizeInMklOrder() argument
490 DCHECK(output_dims_tf_order); in GetOutputAndPadSizeInMklOrder()
511 dilations, output_dims_tf_order, in GetOutputAndPadSizeInMklOrder()
528 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
535 DCHECK(output_dims_tf_order);
548 input_shape, filter_shape, *strides, *dilations, output_dims_tf_order,
|