| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 97 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in Compute()
282 const Tensor& src_tensor = MklGetInput(context, kIdxInput); in SanityCheckInputs()
371 const Tensor& input_grad_tensor = MklGetInput(context, kIdxGradient); in Compute()
372 const Tensor& orig_input_tensor = MklGetInput(context, kIdxOrigInput); in Compute()
421 const Tensor& workspace_tensor = MklGetInput(context, kIdxWorkspace); in Compute()
516 ConvertMklToTF<T>(context, MklGetInput(context, kIdxGradient), in MklDefaultToEigen()
519 input_gradient_tensor = MklGetInput(context, kIdxGradient); in MklDefaultToEigen()
524 context, MklGetInput(context, kIdxOrigInput), in MklDefaultToEigen()
527 orig_input_tensor = MklGetInput(context, kIdxOrigInput); in MklDefaultToEigen()
532 context, MklGetInput(context, kIdxOrigOutput), in MklDefaultToEigen()
[all …]
|
| D | mkl_aggregate_ops.cc | 43 const Tensor& src_tensor = MklGetInput(ctx, src_index); in GetTensorShape()
96 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar()
102 output_tf_shape = MklGetInput(ctx, kOutputIdx).shape(); in ComputeScalar()
138 const Tensor& src_tensor = MklGetInput(ctx, kSrc0Idx); in Compute()
188 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute()
231 output_tf_shape = MklGetInput(ctx, kSrc0Idx).shape(); in Compute()
|
| D | mkl_relu_op.cc | 443 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute()
456 tf_shape_dst = MklGetInput(context, src_index).shape(); in Compute()
594 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute()
595 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute()
613 tf_shape_diff_src = MklGetInput(context, diff_src_index).shape(); in Compute()
773 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
805 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
806 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute_Scalar()
841 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
877 const Tensor& src_tensor = MklGetInput(context, src_index); in Compute_Scalar()
[all …]
|
| D | mkl_dequantize_op.cc | 56 const Tensor& src_tensor = MklGetInput(ctx, kSrcIndex); in Compute()
58 MklGetInput(ctx, kMinIndex).template flat<float>()(0); in Compute()
60 MklGetInput(ctx, kMaxIndex).template flat<float>()(0); in Compute()
|
| D | mkl_maxpooling_op.cc | 57 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
163 const Tensor& min_input_t = MklGetInput(context, 1); in Compute()
164 const Tensor& max_input_t = MklGetInput(context, 2); in Compute()
238 MklGetInput(context, kInputTensorIndexOrigInput); in Compute()
240 MklGetInput(context, kInputTensorIndexGradient); in Compute()
242 MklGetInput(context, kInputTensorIndexWorkspace); in Compute()
|
| D | mkl_avgpooling_op.cc | 53 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
138 const Tensor& min_input_t = MklGetInput(context, 1); in Compute()
139 const Tensor& max_input_t = MklGetInput(context, 2); in Compute()
180 MklGetInput(context, kInputTensorIndexInputShape); in Compute()
182 MklGetInput(context, kInputTensorIndexInputGradient); in Compute()
|
| D | mkl_slice_op.cc | 69 const Tensor& input = MklGetInput(context, kInputTensorIndex); in ValidateMklInputs()
70 const Tensor& begin_tensor = MklGetInput(context, kInputBeginIndex); in ValidateMklInputs()
71 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs()
145 const Tensor& input = MklGetInput(context, 0); in CheckCommonCasesForMklInputs()
399 const Tensor& input_tensor = MklGetInput(context, 0); in ComputeMklSlice()
|
| D | mkl_reshape_op.cc | 62 const Tensor& input_tensor = MklGetInput(context, 0); in Compute()
63 const Tensor& sizes = MklGetInput(context, 1); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 685 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
686 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()
687 const Tensor& shift_tensor = MklGetInput(context, kShiftIndex); in Compute()
688 const Tensor& est_mean_tensor = MklGetInput(context, kMeanIndex); in Compute()
689 const Tensor& est_variance_tensor = MklGetInput(context, kVarianceIndex); in Compute()
929 const Tensor& input = MklGetInput(context, 0); in ExtractParams()
1070 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIndex); in Compute()
1071 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
1072 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()
1073 const Tensor& saved_mean_tensor = MklGetInput(context, kMeanIndex); in Compute()
[all …]
|
| D | mkl_input_conversion_op.cc | 70 const Tensor& input_tensor_0 = MklGetInput(context, kInputIndex_0); in Compute()
74 const Tensor& input_tensor_1 = MklGetInput(context, kInputIndex_1); in Compute()
|
| D | mkl_tfconv_op.h | 69 const Tensor& input_tensor = MklGetInput(context, input_number); in ConvertMklToTf()
|
| D | mkl_matmul_op_fused.cc | 61 const Tensor& bias_tensor = MklGetInput(ctx, this->kInputIndexBias); in Compute()
143 const Tensor& add_tensor = MklGetInput(ctx, kInputIndex_Add); in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 315 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute()
316 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute()
317 const Tensor& diff_dst_tensor = MklGetInput(context, kOutbpropIdx); in Compute()
|
| D | mkl_qmatmul_op.cc | 162 const Tensor& src_tensor = MklGetInput(context, this->kInputIndexSrc); in Compute()
164 MklGetInput(context, this->kInputIndexWeight); in Compute()
165 const Tensor& bias_tensor = MklGetInput(context, this->kInputIndexBias); in Compute()
|
| D | mkl_quantize_op.cc | 267 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in ComputeScalar()
331 const Tensor& src_tensor = MklGetInput(ctx, src_idx); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 372 const Tensor& src_tensor = MklGetInput(context, kInputIdx); in Compute()
373 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute()
374 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIdx); in Compute()
|
| D | mkl_conv_ops.cc | 500 const Tensor& src_tensor = MklGetInput(context, kInputIndex_Src); in Compute()
501 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute()
729 const Tensor& bias_tensor = MklGetInput(context, kInputIndex_Bias); in Compute()
757 const Tensor& paddings_tf = MklGetInput(context, input_index_pad_); in PadWithConvFusion()
877 const Tensor& add_tensor = MklGetInput(context, kInputIndex_Add); in AllocateOutputTensor()
1751 Tensor& summand = const_cast<Tensor&>(MklGetInput(context, summand_idx)); in AllocateOutputTensor()
1780 const Tensor& summand = MklGetInput(context, summand_idx); in AllocateOutputTensor()
|
| D | mkl_softmax_op.cc | 205 const Tensor& src_tensor = MklGetInput(context, src_idx); in Compute()
|
| D | mkl_concat_op.cc | 465 ? MklGetInput(context, 0) in Compute()
466 : MklGetInput(context, N); in Compute()
|
| D | mkl_conv_ops.h | 300 const Tensor& bias = MklGetInput(context_, bias_index); in GetBiasSizeInMklOrder()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 698 inline const Tensor& MklGetInput(OpKernelContext* ctext, int n) { in MklGetInput() function
734 const Tensor& t = MklGetInput(context, input_idx);
|