/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...] |