| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 99 GetMklShape(context, kIdxInput, &src_dnn_shape); in Compute()
284 GetMklShape(context, kIdxInput, &src_dnn_shape); in SanityCheckInputs()
347 GetMklShape(context, kIdxGradient, &input_grad_dnn_shape); in Compute()
348 GetMklShape(context, kIdxOrigInput, &orig_input_dnn_shape); in Compute()
349 GetMklShape(context, kIdxOrigOutput, &orig_output_dnn_shape); in Compute()
509 GetMklShape(context, kIdxGradient, &input_grad_dnn_shape); in MklDefaultToEigen()
510 GetMklShape(context, kIdxOrigInput, &orig_input_dnn_shape); in MklDefaultToEigen()
511 GetMklShape(context, kIdxOrigOutput, &orig_output_dnn_shape); in MklDefaultToEigen()
597 GetMklShape(context, kIdxGradient, &in_grads_dnn_shape); in SanityCheckInputs()
598 GetMklShape(context, kIdxOrigInput, &in_image_dnn_shape); in SanityCheckInputs()
[all …]
|
| D | mkl_aggregate_ops.cc | 45 GetMklShape(ctx, src_index, &src_mkl_shape); in GetTensorShape()
77 GetMklShape(ctx, i, &src_mkl_shape); in FindMKLInputIndex()
121 GetMklShape(ctx, kSrc0Idx, &mkl_shape); in Compute()
170 GetMklShape(ctx, mkl_input_index, &mkl_shape); in Compute()
185 GetMklShape(ctx, src_idx, &src_mkl_shape); in Compute()
|
| D | mkl_relu_op.cc | 445 GetMklShape(context, src_index, &dnn_shape_src); in Compute()
599 GetMklShape(context, src_index, &dnn_shape_src); in Compute()
600 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute()
775 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
810 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
843 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
882 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
924 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
973 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
1009 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
[all …]
|
| D | mkl_slice_op.cc | 74 GetMklShape(context, kInputTensorIndex, &input_mkl_shape); in ValidateMklInputs()
75 GetMklShape(context, kInputBeginIndex, &begin_mkl_shape); in ValidateMklInputs()
76 GetMklShape(context, kInputSizeIndex, &size_mkl_shape); in ValidateMklInputs()
147 GetMklShape(context, 0, &input_mkl_shape); in CheckCommonCasesForMklInputs()
402 GetMklShape(context, 0, &input_mkl_shape); in ComputeMklSlice()
|
| D | mkl_matmul_op_fused.cc | 65 GetMklShape(ctx, this->kInputIndexSrc, &src_mkl_shape, native_format); in Compute()
66 GetMklShape(ctx, this->kInputIndexWeight, &weight_mkl_shape, native_format); in Compute()
145 GetMklShape(ctx, kInputIndex_Add, &add_mkl_shape, native_format); in Compute()
|
| D | mkl_identity_op.cc | 40 GetMklShape(context, kInputIdx, &dnn_shape_input); in Compute()
|
| D | mkl_avgpooling_op.cc | 55 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input, in Compute()
185 GetMklShape(context, kInputTensorIndexInputShape, &orig_input_mkl_shape, in Compute()
187 GetMklShape(context, kInputTensorIndexInputGradient, &grad_mkl_shape, in Compute()
|
| D | mkl_input_conversion_op.cc | 72 GetMklShape(context, kInputIndex_0, &input_shape_0); in Compute()
76 GetMklShape(context, kInputIndex_1, &input_shape_1); in Compute()
|
| D | mkl_maxpooling_op.cc | 59 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input, in Compute()
244 GetMklShape(context, kInputTensorIndexOrigInput, &orig_input_mkl_shape, in Compute()
246 GetMklShape(context, kInputTensorIndexGradient, &grad_mkl_shape, in Compute()
|
| D | mkl_tfconv_op.h | 71 GetMklShape(context, input_number, &input_shape); in ConvertMklToTf()
|
| D | mkl_conv_grad_input_ops.cc | 320 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
321 GetMklShape(context, kFilterIdx, &filter_mkl_shape, native_format); in Compute()
322 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape, native_format); in Compute()
|
| D | mkl_dequantize_op.cc | 64 GetMklShape(ctx, kSrcIndex, &src_mkl_shape); in Compute()
|
| D | mkl_reshape_op.cc | 66 GetMklShape(context, kInputSlotIdx, &mkl_shape_input); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 377 GetMklShape(context, kInputIdx, &src_mkl_shape, native_format); in Compute()
378 GetMklShape(context, kFilterIdx, &filter_mkl_shape, native_format); in Compute()
379 GetMklShape(context, kDiffDstIdx, &diff_dst_mkl_shape, native_format); in Compute()
|
| D | mkl_qmatmul_op.cc | 168 GetMklShape(context, this->kInputIndexSrc, &src_mkl_shape); in Compute()
169 GetMklShape(context, this->kInputIndexWeight, &weight_mkl_shape); in Compute()
|
| D | mkl_softmax_op.cc | 207 GetMklShape(context, src_idx, &src_mkl_shape); in Compute()
|
| D | mkl_conv_ops.cc | 503 GetMklShape(context, kInputIndex_Src, &src_mkl_shape, native_format); in Compute()
504 GetMklShape(context, kInputIndex_Filter, &filter_mkl_shape, in Compute()
879 GetMklShape(context, kInputIndex_Add, &add_mkl_shape, native_format); in AllocateOutputTensor()
1753 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
1785 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
|
| D | mkl_quantize_op.cc | 333 GetMklShape(ctx, src_idx, &src_mkl_shape); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 693 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()
1081 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()
1082 GetMklShape(context, kDiffDstIndex, &dnn_shape_diff_dst, native_format); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 677 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape, in GetMklShape() function
693 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape) { in GetMklShape() function
694 GetMklShape(ctext, n, mklshape, false); in GetMklShape()
730 GetMklShape(context, input_idx, &input_mkl_shape, eager_mode);
|