| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 38 memory::dims src_dims; member in tensorflow::MklSoftmaxParams
42 MklSoftmaxParams(memory::dims src_dims, MklTensorFormat src_fmt, int axis) in MklSoftmaxParams() argument
43 : src_dims(src_dims), src_fmt(src_fmt), axis(axis) {} in MklSoftmaxParams()
118 new memory::desc({fwdParams.src_dims}, MklDnnType<T>(), src_format)); in Setup()
173 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()
215 memory::dims src_dims; in Compute() local
218 src_dims = src_mkl_shape.GetSizesAsMklDnnDims(); in Compute()
221 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
271 MklSoftmaxParams fwdParams(src_dims, src_fmt, axis); in Compute()
288 output_mkl_shape.SetTfLayout(src_dims.size(), src_dims, layout_type); in Compute()
[all …]
|
| D | mkl_relu_op.cc | 44 memory::dims src_dims; member in tensorflow::MklEltwiseFwdParams
50 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams() argument
52 : src_dims(src_dims), in MklEltwiseFwdParams()
197 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()
219 memory::dims src_dims; member in tensorflow::MklEltwiseBwdParams
228 MklEltwiseBwdParams(const memory::dims& src_dims, in MklEltwiseBwdParams() argument
231 : src_dims(src_dims), in MklEltwiseBwdParams()
409 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()
463 memory::dims src_dims; in Compute() local
468 src_dims = dnn_shape_src.GetSizesAsMklDnnDims(); in Compute()
[all …]
|
| D | mkl_quantize_op.cc | 60 memory::dims src_dims; member
70 MklReorderWithScaleFwdParams(memory::dims src_dims, memory::desc src_md, in MklReorderWithScaleFwdParams()
72 : src_dims(src_dims), src_md(src_md), dst_md(dst_md) {} in MklReorderWithScaleFwdParams()
336 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
339 auto output_dims = src_dims; in Compute()
373 : memory::desc(src_dims, MklDnnType<float>(), dst_layout_type); in Compute()
402 memory::desc(src_dims, MklDnnType<T>(), dst_layout_type); in Compute()
468 MklReorderWithScaleFwdParams fwdParams(src_dims, src_md, dst_md); in Compute()
|
| D | mkl_dequantize_op.cc | 68 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
71 auto output_dims = src_dims; in Compute()
86 : memory::desc(src_dims, MklDnnType<T>(), in Compute()
102 dst_md = memory::desc(src_dims, MklDnnType<float>(), in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 44 memory::dims src_dims; member
51 MklBatchNormFwdParams(const memory::dims& src_dims, int depth, float eps, in MklBatchNormFwdParams()
54 : src_dims(src_dims), in MklBatchNormFwdParams()
355 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()
378 memory::dims src_dims; member
387 MklBatchNormBwdParams(memory::dims src_dims, memory::dims diff_dst_dims, in MklBatchNormBwdParams()
390 : src_dims(src_dims), in MklBatchNormBwdParams()
612 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()
767 memory::dims src_dims = in Compute() local
774 : memory::desc(src_dims, MklDnnType<T>(), dnn_fmt); in Compute()
[all …]
|
| D | mkl_aggregate_ops.cc | 194 memory::dims src_dims; in Compute() local
196 src_dims = in Compute()
200 src_dims = TFShapeToMklDnnDimsInNCDHW(src_tensor.shape(), in Compute()
203 md = memory::desc(src_dims, MklDnnType<T>(), dnn_fmt); in Compute()
|
| D | mkl_concat_op.cc | 239 std::vector<memory::dims> src_dims; member
253 src_dims.push_back(src_dims_pt[k]); in MklConcatFwdParams()
421 key_creator.AddAsKey(concat_fwd_dims.src_dims[k]); in CreateKey()
626 memory::dims src_dims(src_md.data.dims, in Compute() local
629 memory::desc(src_dims, MklDnnType<T>(), mkl_common_format); in Compute()
638 memory::dims src_dims = TFShapeToMklDnnDims(s_shape); in Compute() local
639 dst_concat_dim_size += src_dims[concat_dim]; in Compute()
650 memory::desc(src_dims, MklDnnType<T>(), mkl_common_format); in Compute()
656 src_dims_pt.push_back(src_dims); in Compute()
880 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat() local
[all …]
|
| D | mkl_matmul_op_fused.cc | 107 memory::dims src_dims = memory::dims({batch, k}); in Compute() local
120 src_dims, weight_dims, bias_dims, dst_dims, src_format, in Compute()
214 : memory::desc(src_dims, MklDnnType<T>(), src_format); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 56 memory::dims src_dims; member
67 MklConvBwdFilterParams(memory::dims src_dims, memory::dims diff_filter_dims, in MklConvBwdFilterParams()
73 : src_dims(src_dims), in MklConvBwdFilterParams()
215 context_.src_md.reset(new memory::desc({convBwdFilterDims.src_dims}, in Setup()
332 key_creator.AddAsKey(convBwdFilterDims.src_dims); in CreateKey()
|
| D | mkl_pooling_ops_common.h | 41 memory::dims src_dims; member
53 MklPoolingParams(memory::dims src_dims, memory::dims dst_dims, in MklPoolingParams()
59 : src_dims(src_dims), in MklPoolingParams()
189 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()
334 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()
|
| D | mkl_avgpooling_op.cc | 90 memory::dims src_dims = in Compute() local
99 : memory::desc(src_dims, MklDnnType<T>(), in Compute()
114 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
|
| D | mkl_qmatmul_op.cc | 177 memory::dims src_dims, weight_dims; in Compute() local
188 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
212 : memory::desc(src_dims, MklDnnType<Tinput>(), input_output_fmt); in Compute()
227 MklDnnMatMulFwdParams matmul_fwd_dims(src_dims, weight_dims, bias_dims, in Compute()
|
| D | mkl_matmul_ops_common.h | 47 memory::dims src_dims; member
61 MklDnnMatMulFwdParams(memory::dims src_dims, memory::dims weight_dims,
66 : src_dims(src_dims), in src_dims() function
179 context_.src_md.reset(new memory::desc({matmul_fwd_params.src_dims}, in Setup()
336 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.src_dims); in CreateKey()
|
| D | mkl_conv_ops.cc | 59 memory::dims src_dims; member
78 MklConvFwdParams(memory::dims src_dims, memory::dims filter_dims, in MklConvFwdParams()
83 : src_dims(src_dims), in MklConvFwdParams()
218 {convFwdDims.src_dims}, MklDnnType<Tinput>(), user_data_fmt)); in Setup()
365 key_creator.AddAsKey(convFwdDims.src_dims); in CreateKey()
514 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
541 src_tf_shape, filter_tf_shape, &src_dims, &filter_dims, &strides, in Compute()
614 : memory::desc(src_dims, MklDnnType<Tinput>(), mkl_fmt_tag); in Compute()
642 (src_dims[MklDnnDims::Dim_N] > kSmallBatchSize) && in Compute()
654 src_dims, filter_dims, fuse_biasadd_ ? bias_dims : NONE_DIMS, in Compute()
|
| D | mkl_maxpooling_op.cc | 119 memory::dims src_dims = in Compute() local
140 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
|
| D | mkl_pooling_ops_common.cc | 138 context_.src_md.reset(new memory::desc({bwdParams.src_dims}, MklDnnType<T>(), in Setup()
|