| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.h | 95 TensorFormat data_format, const MklDnnShape* mkl_in_shape);
155 const MklDnnShape& original_input_mkl_shape,
195 const MklDnnShape& input_mkl_shape,
221 MklDnnShape output_mkl_shape;
254 const MklDnnShape& input_mkl_shape) {
288 MklDnnShape output_mkl_shape;
331 const MklDnnShape& original_input_mkl_shape,
353 const MklDnnShape& original_output_mkl_shape,
364 const MklDnnShape& input_gradient_mkl_shape,
|
| mkl_softmax_op.cc | 58 MklDnnShape src_mkl_shape;
101 MklDnnShape output_mkl_shape;
|
| mkl_identity_op.cc | 68 MklDnnShape dnn_shape_input;
|
| mkl_relu_op.cc | 388 MklDnnShape dnn_shape_src;
423 MklDnnShape dnn_shape_dst;
487 MklDnnShape dnn_shape_src, dnn_shape_diff_dst;
521 // in Mkl format, we just get Mkl layout from MklDnnShape. For input in
580 MklDnnShape dnn_shape_diff_src;
643 MklDnnShape dnn_shape_src;
649 MklDnnShape dnn_shape_dst;
676 MklDnnShape dnn_shape_diff_dst;
679 MklDnnShape dnn_shape_diff_src;
706 MklDnnShape dnn_shape_src
[all...] |
| mkl_input_conversion_op.cc | 267 MklDnnShape input_shape_0;
271 MklDnnShape input_shape_1;
314 // Create MklDnnShape for output mkl tensor based on input0
316 MklDnnShape mkl_output_mkl_shape;
378 const MklDnnShape* mkl_shape;
380 MklDnnShape* tf_mkl_shape;
422 // Create MklDnnShape for output Mkl tensor.
424 MklDnnShape mkl_output_mkl_shape;
|
| mkl_reshape_op.cc | 145 bool SkipReorder(const MklDnnShape& mkl_shape_input,
166 MklDnnShape mkl_shape_input;
237 // need to update MklDnnShape object associated with the input
275 // update MklDnnShape object for the tensorflow tensor, and forward
284 // Set MklDnnShape
285 MklDnnShape mkl_shape_output;
293 // output MklDnnShape object.
|
| mkl_maxpooling_op.cc | 515 MklDnnShape dnn_shape_input;
595 MklDnnShape workspace_mkl_shape;
630 MklDnnShape orig_input_mkl_shape, orig_output_mkl_shape, grad_mkl_shape,
724 const MklDnnShape& original_input_mkl_shape,
746 const MklDnnShape& orig_input_mkl_shape,
747 const MklDnnShape& orig_output_mkl_shape,
748 const MklDnnShape& grad_mkl_shape,
749 const MklDnnShape& workspace_mkl_shape) {
[all...] |
| mkl_conv_grad_filter_ops.cc | 447 void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
448 const MklDnnShape& filter_mkl_shape,
449 const MklDnnShape& obp_mkl_shape) {
579 MklDnnShape output_mkl_shape;
597 MklDnnShape bias_grad_mkl_shape;
|
| mkl_conv_ops.h | 368 MklDnnShape input_mkl_shape, filter_mkl_shape, outbprop_mkl_shape;
389 MklDnnShape output_mkl_shape;
503 virtual void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
504 const MklDnnShape& filter_mkl_shape,
505 const MklDnnShape& outbprop_mkl_shape) = 0;
|
| mkl_fused_batch_norm_op.cc | [all...] |
| mkl_conv_grad_input_ops.cc | 372 void ValidateMklShapes(const MklDnnShape& input_mkl_shape,
373 const MklDnnShape& filter_mkl_shape,
374 const MklDnnShape& obp_mkl_shape) {
464 MklDnnShape output_mkl_shape;
|
| mkl_pooling_ops_common.cc | 66 const MklDnnShape* mklInputShape) {
|
| mkl_tfconv_op.h | 71 MklDnnShape input_shape;
|
| mkl_lrn_op.cc | 766 MklDnnShape src_dnn_shape;
[all...] |
| mkl_avgpooling_op.cc | 448 MklDnnShape dnn_shape_input;
470 MklDnnShape output_mkl_shape;
551 MklDnnShape original_input_mkl_shape, input_gradient_mkl_shape;
652 const MklDnnShape& original_input_mkl_shape,
673 const MklDnnShape& original_input_mkl_shape,
674 const MklDnnShape& input_gradient_mkl_shape) {
|
| mkl_conv_ops.cc | 520 MklDnnShape src_mkl_shape, filter_mkl_shape;
552 MklDnnShape output_mkl_shape;
674 MklDnnShape output_mkl_shape;
698 MklDnnShape filter_mkl_shape;
[all...] |
| mkl_aggregate_ops.cc | 309 MklDnnShape src1_mkl_shape, src2_mkl_shape;
388 // in Mkl format, we just get Mkl layout from MklDnnShape. For input in
457 MklDnnShape output_mkl_shape;
|
| mkl_concat_op.cc | 588 std::vector<MklDnnShape> input_shapes(N);
747 MklDnnShape dnn_shape_dst;
796 MklDnnShape dnn_shape_output;
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util_test.cc | 33 MklDnnShape a_mkldnn_shape;
45 MklDnnShape b_mkldnn_shape;
|
| mkl_util.h | 337 class MklDnnShape {
359 MklDnnShape() {
369 ~MklDnnShape() {}
370 TF_DISALLOW_COPY_AND_ASSIGN(MklDnnShape); // Cannot copy
390 /// Equality function for MklDnnShape objects
392 inline bool operator==(const MklDnnShape& input_shape) const {
408 /// Equality operator for MklDnnShape and TFShape.
455 // MklDnnShape object.
664 typedef std::vector<MklDnnShape> MklDnnShapeList;
705 const MklDnnShape& mkl_shape)
[all...] |
