| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_ops.h | 283 auto filter_tf_shape = GetTfShape(context_, filter_index); local
289 GetOutputAndPadSizeInMklOrder(input_tf_shape, filter_tf_shape, strides,
381 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor); variable
387 filter_tf_shape.num_elements() == 0 ||
392 input_tf_shape, filter_tf_shape, outbprop_tf_shape);
415 input_tf_shape, filter_tf_shape, &fwd_input_dims, &fwd_filter_dims,
|
| mkl_conv_ops.cc | 537 auto filter_tf_shape = GetTfShape(context, kInputIndex_Filter); variable
539 src_tf_shape, filter_tf_shape, &src_dims, &filter_dims, &strides,
561 &output_filter_tensor, filter_tf_shape,
621 TFShapeToMklDnnDims(filter_tf_shape),
642 TFShapeToMklDnnDims(filter_tf_shape),
710 TensorShape filter_tf_shape; local
711 filter_tf_shape.AddDim((filter_pd.get_size() / sizeof(T)));
714 filter_tf_shape, filter_mkl_shape);
[all...] |
| mkl_conv_grad_filter_ops.cc | 464 TensorShape filter_tf_shape; local
467 &filter_tf_shape)
470 return filter_tf_shape;
|
