| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_relu_op.cc | 457 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, tf_shape_dst, in Compute()
522 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, tf_shape_dst, in Compute()
532 AllocateOutputSetMklShape(context, dst_index, dnn_shape_dst); in Compute()
614 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute()
734 AllocateOutputSetMklShape(context, diff_src_index, dnn_shape_diff_src); in Compute()
782 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
814 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
850 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
886 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
931 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
[all …]
|
| D | mkl_quantize_op.cc | 273 AllocateOutputSetMklShape(ctx, 0, &output_tensor, src_tensor.shape(), in ComputeScalar()
279 AllocateOutputSetMklShape(ctx, 1, &output_min_tensor, min_tf_shape, in ComputeScalar()
285 AllocateOutputSetMklShape(ctx, 2, &output_max_tensor, max_tf_shape, in ComputeScalar()
421 AllocateOutputSetMklShape(ctx, 0, &output_tensor, output_tf_shape, in Compute()
429 AllocateOutputSetMklShape(ctx, 1, &output_min_tensor, min_tf_shape, in Compute()
435 AllocateOutputSetMklShape(ctx, 2, &output_max_tensor, max_tf_shape, in Compute()
|
| D | mkl_aggregate_ops.cc | 103 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in ComputeScalar()
146 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in Compute()
233 AllocateOutputSetMklShape(ctx, kOutputIdx, &dst_tensor, output_tf_shape, in Compute()
|
| D | mkl_concat_op.cc | 730 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
765 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
787 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
789 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
803 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
850 AllocateOutputSetMklShape(context, 0, output_data_mkl_shape); in CallEigenVersion()
854 AllocateOutputSetMklShape(context, 1, output_min_max_mkl_shape); in CallEigenVersion()
855 AllocateOutputSetMklShape(context, 2, output_min_max_mkl_shape); in CallEigenVersion()
|
| D | mkl_fused_batch_norm_op.cc | 859 AllocateOutputSetMklShape(context, kDstIndex, &dst_tensor, tf_shape_dst, in Compute()
947 AllocateOutputSetMklShape(context, kDstIndex, dst_tensor, tf_shape_src, in HandleEmptyInput()
985 AllocateOutputSetMklShape(context, kBatchMeanIndex, batch_mean_tensor, in AllocateTFOutputs()
998 AllocateOutputSetMklShape(context, kBatchVarianceIndex, in AllocateTFOutputs()
1010 AllocateOutputSetMklShape(context, kSavedMeanIndex, saved_mean_tensor, in AllocateTFOutputs()
1021 AllocateOutputSetMklShape(context, kSavedVarianceIndex, in AllocateTFOutputs()
1036 AllocateOutputSetMklShape(context, kReservedSpaceIndex, in AllocateTFOutputs()
1256 AllocateOutputSetMklShape(context, kDiffSrcIndex, &diff_src_tensor, in Compute()
1323 AllocateOutputSetMklShape(context, kDiffSrcIndex, diff_src_tensor, in HandleEmptyInput()
1350 AllocateOutputSetMklShape(context, kDiffScaleIndex, diff_scale_tensor, in AllocateTFOutputs()
[all …]
|
| D | mkl_lrn_op.cc | 210 AllocateOutputSetMklShape(context, kIdxOutput, output_tensor, in AllocateOutputTensor()
234 AllocateOutputSetMklShape(context, kIdxOutput, &output_dnn_data, in MklDefaultToEigen()
243 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor, in MklDefaultToEigen()
275 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor, in AllocateWorkspaceTensor()
472 AllocateOutputSetMklShape(context, kIdxOutput, output_tensor, in AllocateOutputTensor()
554 AllocateOutputSetMklShape(context, kIdxOutput, &output_dnn_data, in MklDefaultToEigen()
|
| D | mkl_conv_grad_filter_ops.cc | 403 AllocateOutputSetMklShape(context, kOutputIdx, &diff_filter_tensor, in Compute()
493 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
514 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
526 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
693 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
704 AllocateOutputSetMklShape(context, 1, bias_grad_tensor, bias_grad_shape, in AllocateBiasGradTensor()
|
| D | mkl_input_conversion_op.cc | 134 AllocateOutputSetMklShape(context, kInputIndex_0, &tensor_out, in Compute()
235 AllocateOutputSetMklShape(context, tf_tensor_index, &tensor_out, in Compute()
|
| D | mkl_maxpooling_op.cc | 173 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
175 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
214 AllocateOutputSetMklShape(context, kOutputTensorIndexWorkspace, in AllocateWorkspaceTensor()
|
| D | mkl_slice_op.cc | 154 AllocateOutputSetMklShape(context, 0, input_mkl_shape); in CheckCommonCasesForMklInputs()
392 AllocateOutputSetMklShape(context, 0, &output_tensor, shape_to, in ComputeMklSlice()
498 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
|
| D | mkl_matmul_op_fused.cc | 159 AllocateOutputSetMklShape(ctx, kOutputIndex_Dst, &dst_tensor, in Compute()
192 AllocateOutputSetMklShape(ctx, 0, &dst_tensor, output_tf_shape, in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 355 AllocateOutputSetMklShape(context, kOutputIdx, &diff_src_tensor, in Compute()
454 AllocateOutputSetMklShape(context, 0, &diff_src_tensor, diff_src_tf_shape, in Compute()
581 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 560 AllocateOutputSetMklShape(context, kOutputIndex_Dst, &dst_tensor, in Compute()
568 AllocateOutputSetMklShape(context, kOutputIndex_Filter, in Compute()
894 AllocateOutputSetMklShape(context, kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
927 AllocateOutputSetMklShape(context, kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
1033 AllocateOutputSetMklShape(context, kOutputIndex_Filter, filter_tensor, in AllocateFilterOutputTensor()
1397 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
1399 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
1415 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
1417 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
1423 AllocateOutputSetMklShape(context, 1, &output_min, in Compute()
[all …]
|
| D | mkl_avgpooling_op.cc | 148 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
150 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
|
| D | mkl_dequantize_op.cc | 122 AllocateOutputSetMklShape(ctx, 0, &output_tensor, output_tf_shape, in Compute()
|
| D | mkl_reshape_op.cc | 163 AllocateOutputSetMklShape(context, kOutputSlotIdx, &output_tensor, in Compute()
|
| D | mkl_pooling_ops_common.h | 570 AllocateOutputSetMklShape(context, kOutputIndex, output_tensor, in AllocateEmptyOutputTensor()
663 AllocateOutputSetMklShape(context, kOutputTensorIndexOutput, output_tensor, in AllocateOutputTensor()
715 AllocateOutputSetMklShape(context, kOutputTensorIndexOutput, output_tensor, in AllocateOutputTensor()
|
| D | mkl_qmatmul_op.cc | 327 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
329 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
|
| D | mkl_softmax_op.cc | 295 AllocateOutputSetMklShape(context, 0, &output_tensor, output_tf_shape, in Compute()
|
| D | mkl_matmul_ops_common.h | 408 AllocateOutputSetMklShape(context, kOutputIndexDst, output_tensor, in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 741 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n, in AllocateOutputSetMklShape() function
756 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n,
845 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()
861 AllocateOutputSetMklShape(context, idx_out, dnn_shape_output); in ForwardTfTensorInToOut()
892 AllocateOutputSetMklShape(context, idx_data_out, mkl_shape_output); in SetDummyMklDnnShapeOutput()
921 AllocateOutputSetMklShape(context, idx_out, mkl_shape);
|