| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_lrn_op.cc | 95 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in Compute()
289 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in SanityCheckInputs()
375 const Tensor& input_grad_tensor = MklGetInput(context, kIdxGradient); in Compute()
376 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in Compute()
423 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); in Compute()
535 context, MklGetInput(context, kIdxGradient), input_grad_dnn_shape); in MklDefaultToEigen()
537 input_gradient_tensor = MklGetInput(context, kIdxGradient); in MklDefaultToEigen()
542 context, MklGetInput(context, kIdxOrigInput), orig_input_dnn_shape); in MklDefaultToEigen()
544 orig_input_tensor = MklGetInput(context, kIdxOrigInput); in MklDefaultToEigen()
549 context, MklGetInput(context, kIdxOrigOutput), orig_output_dnn_shape); in MklDefaultToEigen()
[all …]
|
| D | mkl_relu_op.cc | 460 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute()
571 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute()
572 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute()
730 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
760 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
761 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute_Scalar()
793 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
827 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
828 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute_Scalar()
865 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
[all …]
|
| D | mkl_avgpooling_op.cc | 70 const Tensor& tensor_in = MklGetInput(context, 0); in Compute()
245 const Tensor& tensor_in_shape = MklGetInput(context, 0); in Compute()
246 const Tensor& out_backprop = MklGetInput(context, 1); in Compute()
353 const Tensor& tensor_in_shape = MklGetInput(context, 0); in MklCreateLayoutsAndPrimitives()
354 const Tensor& out_backprop = MklGetInput(context, 1); in MklCreateLayoutsAndPrimitives()
446 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
537 const Tensor& min_input_t = MklGetInput(context, 1); in Compute()
538 const Tensor& max_input_t = MklGetInput(context, 2); in Compute()
577 MklGetInput(context, kInputTensorIndexInputShape); in Compute()
579 MklGetInput(context, kInputTensorIndexInputGradient); in Compute()
|
| D | mkl_maxpooling_op.cc | 79 const Tensor& tensor_in = MklGetInput(context, 0); in Compute()
237 const Tensor& tensor_in = MklGetInput(context, 0); in Compute()
391 const Tensor& tensor_in = MklGetInput(context, 0); in MklPrepareInputs()
392 const Tensor& out_backprop = MklGetInput(context, 2); in MklPrepareInputs()
442 const Tensor& workspace = MklGetInput(context, 3); in MklPrepareInputs()
515 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
614 const Tensor& min_input_t = MklGetInput(context, 1); in Compute()
615 const Tensor& max_input_t = MklGetInput(context, 2); in Compute()
690 MklGetInput(context, kInputTensorIndexOrigInput); in Compute()
692 MklGetInput(context, kInputTensorIndexGradient); in Compute()
[all …]
|
| D | mkl_fused_batch_norm_op.cc | 518 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
519 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()
520 const Tensor& shift_tensor = MklGetInput(context, kShiftIndex); in Compute()
521 const Tensor& est_mean_tensor = MklGetInput(context, kMeanIndex); in Compute()
522 const Tensor& est_variance_tensor = MklGetInput(context, kVarianceIndex); in Compute()
726 const Tensor& input = MklGetInput(context, 0); in ExtractParams()
841 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIndex); in Compute()
842 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
843 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()
844 const Tensor& saved_mean_tensor = MklGetInput(context, kMeanIndex); in Compute()
[all …]
|
| D | mkl_slice_op.cc | 72 const Tensor& input = MklGetInput(context, kInputTensorIndex); in ValidateMklInputs()
73 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs()
74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs()
147 const Tensor& input = MklGetInput(context, 0); in CheckCommonCasesForMklInputs()
391 const Tensor& input_tensor = MklGetInput(context, 0); in ComputeMklSlice()
|
| D | mkl_reshape_op.cc | 65 const Tensor& input_tensor = MklGetInput(context, 0); in Compute()
66 const Tensor& sizes = MklGetInput(context, 1); in Compute()
|
| D | mkl_input_conversion_op.cc | 72 const Tensor& input_tensor_0 = MklGetInput(context, kInputIndex_0); in Compute()
76 const Tensor& input_tensor_1 = MklGetInput(context, kInputIndex_1); in Compute()
|
| D | mkl_conv_ops.cc | 442 const Tensor& input = MklGetInput(context, 0); in Compute()
446 const Tensor& filter = MklGetInput(context, 1); in Compute()
453 const Tensor& bias = MklGetInput(context, 2); in Compute()
601 const Tensor& bias = MklGetInput(context, 2); in Compute()
748 const Tensor& input = MklGetInput(context, 0); in MklPrepareConvolutionInputs()
772 const Tensor& filter = MklGetInput(context, 1); in MklPrepareConvolutionInputs()
800 const Tensor& bias = MklGetInput(context, 2); in MklPrepareConvolutionInputs()
923 const Tensor& src_tensor = MklGetInput(context, kInputIndex_Src); in Compute()
924 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute()
1147 const Tensor& bias_tensor = MklGetInput(context, kInputIndex_Bias); in Compute()
[all …]
|
| D | mkl_conv_grad_bias_ops.cc | 66 const Tensor& input = MklGetInput(context, 0); in Compute()
215 const Tensor& input = MklGetInput(context, 0); in MklPrepareConvolutionInputs()
|
| D | mkl_aggregate_ops.cc | 52 const Tensor& src1_tensor = MklGetInput(ctx, src1_idx); in Compute()
53 const Tensor& src2_tensor = MklGetInput(ctx, src2_idx); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 317 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute()
318 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute()
319 const Tensor& diff_dst_tensor = MklGetInput(context, kOutbpropIdx); in Compute()
|
| D | mkl_tfconv_op.h | 66 const Tensor& input_tensor = MklGetInput(context, input_number); in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 380 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute()
381 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute()
382 const Tensor& diff_dst_tensor = MklGetInput(context, kOutbpropIdx); in Compute()
|
| D | mkl_softmax_op.cc | 52 const Tensor& src_tensor = MklGetInput(context, src_idx); in Compute()
|
| D | mkl_concat_op.cc | 167 ? MklGetInput(context, 0) in Compute()
168 : MklGetInput(context, N); in Compute()
|
| D | mkl_conv_ops.h | 294 const Tensor& bias = MklGetInput(context_, bias_index); in GetBiasSizeInMklOrder()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 900 inline const Tensor& MklGetInput(OpKernelContext* ctext, int n) { in MklGetInput() function
954 const Tensor& t = MklGetInput(context, input_idx); in GetTfShape()
|