| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_lrn_op.cc | 97 GetMklShape(context, kIdxInput, &src_dnn_shape); in Compute()
291 GetMklShape(context, kIdxInput, &src_dnn_shape); in SanityCheckInputs()
351 GetMklShape(context, kIdxGradient, &input_grad_dnn_shape); in Compute()
352 GetMklShape(context, kIdxOrigInput, &orig_input_dnn_shape); in Compute()
353 GetMklShape(context, kIdxOrigOutput, &orig_output_dnn_shape); in Compute()
529 GetMklShape(context, kIdxGradient, &input_grad_dnn_shape); in MklDefaultToEigen()
530 GetMklShape(context, kIdxOrigInput, &orig_input_dnn_shape); in MklDefaultToEigen()
531 GetMklShape(context, kIdxOrigOutput, &orig_output_dnn_shape); in MklDefaultToEigen()
613 GetMklShape(context, kIdxGradient, &in_grads_dnn_shape); in SanityCheckInputs()
614 GetMklShape(context, kIdxOrigInput, &in_image_dnn_shape); in SanityCheckInputs()
[all …]
|
| D | mkl_relu_op.cc | 462 GetMklShape(context, src_index, &dnn_shape_src); in Compute()
576 GetMklShape(context, src_index, &dnn_shape_src); in Compute()
577 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute()
732 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
765 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
795 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
832 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
867 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
903 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst); in Compute_Scalar()
939 GetMklShape(context, src_index, &dnn_shape_src); in Compute_Scalar()
[all …]
|
| D | mkl_slice_op.cc | 77 GetMklShape(context, kInputTensorIndex, &input_mkl_shape); in ValidateMklInputs()
78 GetMklShape(context, kInputBeginIndex, &begin_mkl_shape); in ValidateMklInputs()
79 GetMklShape(context, kInputSizeIndex, &size_mkl_shape); in ValidateMklInputs()
149 GetMklShape(context, 0, &input_mkl_shape); in CheckCommonCasesForMklInputs()
393 GetMklShape(context, 0, &input_mkl_shape); in ComputeMklSlice()
|
| D | mkl_avgpooling_op.cc | 71 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
247 GetMklShape(context, 1, &mkl_context.out_backprop_shape); in Compute()
448 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input); in Compute()
582 GetMklShape(context, kInputTensorIndexInputShape, &orig_input_mkl_shape); in Compute()
583 GetMklShape(context, kInputTensorIndexInputGradient, &grad_mkl_shape); in Compute()
|
| D | mkl_maxpooling_op.cc | 80 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
242 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
243 GetMklShape(context, 2, &mkl_context.output_backprop_shape); in Compute()
517 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input); in Compute()
696 GetMklShape(context, kInputTensorIndexOrigInput, &orig_input_mkl_shape); in Compute()
697 GetMklShape(context, kInputTensorIndexGradient, &grad_mkl_shape); in Compute()
|
| D | mkl_input_conversion_op.cc | 74 GetMklShape(context, kInputIndex_0, &input_shape_0); in Compute()
78 GetMklShape(context, kInputIndex_1, &input_shape_1); in Compute()
|
| D | mkl_identity_op.cc | 41 GetMklShape(context, kInputIdx, &dnn_shape_input); in Compute()
|
| D | mkl_aggregate_ops.cc | 56 GetMklShape(ctx, src1_idx, &src1_mkl_shape); in Compute()
57 GetMklShape(ctx, src2_idx, &src2_mkl_shape); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 322 GetMklShape(context, kInputIdx, &src_mkl_shape); in Compute()
323 GetMklShape(context, kFilterIdx, &filter_mkl_shape); in Compute()
324 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape); in Compute()
|
| D | mkl_tfconv_op.h | 68 GetMklShape(context, input_number, &input_shape); in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 385 GetMklShape(context, kInputIdx, &src_mkl_shape); in Compute()
386 GetMklShape(context, kFilterIdx, &filter_mkl_shape); in Compute()
387 GetMklShape(context, kOutbpropIdx, &diff_dst_mkl_shape); in Compute()
|
| D | mkl_softmax_op.cc | 55 GetMklShape(context, src_idx, &src_mkl_shape); in Compute()
|
| D | mkl_reshape_op.cc | 69 GetMklShape(context, kInputSlotIdx, &mkl_shape_input); in Compute()
|
| D | mkl_conv_grad_bias_ops.cc | 67 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
|
| D | mkl_conv_ops.cc | 443 GetMklShape(context, 0, &(mkl_context.input_shape)); in Compute()
448 GetMklShape(context, 1, &mkl_filter_shape); in Compute()
931 GetMklShape(context, kInputIndex_Src, &src_mkl_shape); in Compute()
932 GetMklShape(context, kInputIndex_Filter, &filter_mkl_shape); in Compute()
1770 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
1801 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
|
| D | mkl_fused_batch_norm_op.cc | 526 GetMklShape(context, kSrcIndex, &dnn_shape_src); in Compute()
849 GetMklShape(context, kSrcIndex, &dnn_shape_src); in Compute()
850 GetMklShape(context, kDiffDstIndex, &dnn_shape_diff_dst); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 876 inline void GetMklShape(OpKernelContext* ctext, int n, MklShape* mklshape) { in GetMklShape() function
887 inline void GetMklShape(OpKernelContext* ctext, int n, MklDnnShape* mklshape) { in GetMklShape() function
950 GetMklShape(context, input_idx, &input_mkl_shape); in GetTfShape()
|