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

  /external/tensorflow/tensorflow/core/kernels/
mkl_concat_op.cc 394 TensorShape mkl_output_tf_shape; local
395 mkl_output_tf_shape.AddDim(1);
396 mkl_output_tf_shape.AddDim(
402 AllocateOutputSetMklShape(context, 0, &output, mkl_output_tf_shape,
    [all...]
mkl_lrn_op.cc 216 TensorShape mkl_output_tf_shape; /* First tensor */ local
225 mkl_output_tf_shape.AddDim(
230 mkl_output_tf_shape /* First tensor */,
447 TensorShape mkl_output_tf_shape; /* First tensor */ variable
461 mkl_output_tf_shape.AddDim(
466 AllocateOutputSetMklShape(context, 0, &output, mkl_output_tf_shape,
    [all...]
mkl_conv_ops.cc 274 TensorShape mkl_output_tf_shape; variable
284 mkl_output_tf_shape.AddDim(
288 AllocateOutputSetMklShape(context, 0, &output, mkl_output_tf_shape,
    [all...]

Completed in 6281 milliseconds