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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h 216 const memory::dims& strides, memory::dims* output_dims_tf_order,
219 CHECK_NOTNULL(output_dims_tf_order);
254 *output_dims_tf_order = TFShapeToMklDnnDims(out_shape);
275 memory::dims* output_dims_tf_order, memory::dims* output_dims_mkl_order,
277 CHECK_NOTNULL(output_dims_tf_order);
290 output_dims_tf_order, output_dims_mkl_order,
303 memory::dims* strides, memory::dims* output_dims_tf_order,
309 CHECK_NOTNULL(output_dims_tf_order);
320 output_dims_tf_order, output_dims_mkl_order,
mkl_conv_ops.cc 532 memory::dims output_dims_tf_order, output_dims_mkl_order; variable
540 &output_dims_tf_order, &output_dims_mkl_order, &padding_l,
545 TensorShape output_tf_shape = MklDnnDimsToTFShape(output_dims_tf_order);
549 if (output_tf_shape.num_elements() == 0 || output_dims_tf_order[0] == 0) {
    [all...]

Completed in 347 milliseconds