| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_batch_norm_op.cc | 669 AllocateOutputSetMklShape(context, kDstIndex, &dst_tensor, tf_shape_dst, in Compute()
743 AllocateOutputSetMklShape(context, kDstIndex, dst_tensor, tf_shape_src, in HandleEmptyInput()
776 AllocateOutputSetMklShape(context, kBatchMeanIndex, batch_mean_tensor, in AllocateTFOutputs()
787 AllocateOutputSetMklShape(context, kBatchVarianceIndex, in AllocateTFOutputs()
799 AllocateOutputSetMklShape(context, kSavedMeanIndex, saved_mean_tensor, in AllocateTFOutputs()
808 AllocateOutputSetMklShape(context, kSavedVarianceIndex, in AllocateTFOutputs()
1001 AllocateOutputSetMklShape(context, kDiffSrcIndex, &diff_src_tensor, in Compute()
1059 AllocateOutputSetMklShape(context, kDiffSrcIndex, diff_src_tensor, in HandleEmptyInput()
1085 AllocateOutputSetMklShape(context, kDiffScaleIndex, diff_scale_tensor, in AllocateTFOutputs()
1094 AllocateOutputSetMklShape(context, kDiffShiftIndex, diff_shift_tensor, in AllocateTFOutputs()
[all …]
|
| D | mkl_relu_op.cc | 526 AllocateOutputSetMklShape(context, dst_index, dnn_shape_dst); in Compute()
694 AllocateOutputSetMklShape(context, diff_src_index, dnn_shape_diff_src); in Compute()
739 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
769 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
802 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
836 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
874 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
907 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
945 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, in Compute_Scalar()
977 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
[all …]
|
| D | mkl_aggregate_ops.cc | 84 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, in Compute()
98 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, in Compute()
236 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, output_tf_shape, in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 410 AllocateOutputSetMklShape(context, kOutputIdx, &diff_filter_tensor, in Compute()
495 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
515 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
527 AllocateOutputSetMklShape(context, 0, &diff_filter_tensor, in Compute()
685 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
697 AllocateOutputSetMklShape(context, 1, bias_grad_tensor, bias_grad_shape, in AllocateBiasGradTensor()
|
| D | mkl_lrn_op.cc | 215 AllocateOutputSetMklShape(context, kIdxOutput, output_tensor, in AllocateOutputTensor()
240 AllocateOutputSetMklShape(context, kIdxOutput, &output_dnn_data, in MklDefaultToEigen()
249 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor, in MklDefaultToEigen()
282 AllocateOutputSetMklShape(context, kIdxWorkspace, &workspace_tensor, in AllocateWorkspaceTensor()
465 AllocateOutputSetMklShape(context, kIdxOutput, output_tensor, in AllocateOutputTensor()
570 AllocateOutputSetMklShape(context, kIdxOutput, &output_dnn_data, in MklDefaultToEigen()
|
| D | mkl_maxpooling_op.cc | 118 AllocateOutputSetMklShape(context, 0, &output_tensor, tensor_out_shape, in Compute()
129 AllocateOutputSetMklShape(context, 1, &workspace_tensor, workspace_shape, in Compute()
291 AllocateOutputSetMklShape(context, 0, &output_tensor, mkl_input_backprop, in Compute()
624 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
626 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
667 AllocateOutputSetMklShape(context, kOutputTensorIndexWorkspace, in AllocateWorkspaceTensor()
|
| D | mkl_input_conversion_op.cc | 135 AllocateOutputSetMklShape(context, kInputIndex_0, &tensor_out, in Compute()
233 AllocateOutputSetMklShape(context, tf_tensor_index, &tensor_out, in Compute()
|
| D | mkl_avgpooling_op.cc | 137 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape, in Compute()
328 AllocateOutputSetMklShape(context, 0, &output, tensor_out_shape, in Compute()
547 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
549 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
|
| D | mkl_slice_op.cc | 156 AllocateOutputSetMklShape(context, 0, input_mkl_shape); in CheckCommonCasesForMklInputs()
384 AllocateOutputSetMklShape(context, 0, &output_tensor, shape_to, in ComputeMklSlice()
472 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 544 AllocateOutputSetMklShape(context, 0, &output, input.shape(), in Compute()
553 AllocateOutputSetMklShape(context, 0, &output, input.shape(), in Compute()
639 AllocateOutputSetMklShape(context, 0, &output, mkl_output_tf_shape, in Compute()
661 AllocateOutputSetMklShape(context, 1, &mkl_context.output_filter, in Compute()
979 AllocateOutputSetMklShape(context, kOutputIndex_Dst, &dst_tensor, in Compute()
989 AllocateOutputSetMklShape(context, kOutputIndex_Filter, in Compute()
1275 AllocateOutputSetMklShape(context, kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
1351 AllocateOutputSetMklShape(context, kOutputIndex_Filter, filter_tensor, in AllocateFilterOutputTensor()
1569 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
1571 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 347 AllocateOutputSetMklShape(context, kOutputIdx, &diff_src_tensor, in Compute()
432 AllocateOutputSetMklShape(context, 0, &diff_src_tensor, diff_src_tf_shape, in Compute()
555 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape, in AllocateOutputTensor()
|
| D | mkl_concat_op.cc | 409 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
429 AllocateOutputSetMklShape(context, 1, &output_min, {}, in Compute()
431 AllocateOutputSetMklShape(context, 2, &output_max, {}, in Compute()
|
| D | mkl_softmax_op.cc | 156 AllocateOutputSetMklShape(context, 0, &output_tensor, output_tf_shape, in Compute()
|
| D | mkl_pooling_ops_common.h | 571 AllocateOutputSetMklShape(context, kOutputIndex, output_tensor, in AllocateEmptyOutputTensor()
658 AllocateOutputSetMklShape(context, kOutputTensorIndexOutput, output_tensor, in AllocateOutputTensor()
709 AllocateOutputSetMklShape(context, kOutputTensorIndexOutput, output_tensor, in AllocateOutputTensor()
|
| D | mkl_reshape_op.cc | 158 AllocateOutputSetMklShape(context, kOutputSlotIdx, &output_tensor, in Compute()
|
| D | mkl_conv_grad_bias_ops.cc | 94 AllocateOutputSetMklShape(context, 0, &bias_backprop, output_shape, in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 963 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n, in AllocateOutputSetMklShape() function
979 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n, in AllocateOutputSetMklShape() function
996 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n, in AllocateOutputSetMklShape() function
1017 inline void AllocateOutputSetMklShape(OpKernelContext* ctext, int n, in AllocateOutputSetMklShape() function
1173 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()
1191 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()
1210 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in ForwardTfTensorInToOut()
1229 AllocateOutputSetMklShape(context, idx_out, dnn_shape_output); in ForwardTfTensorInToOut()
1263 AllocateOutputSetMklShape(context, idx_data_out, mkl_shape_output); in SetDummyMklDnnShapeOutput()
1274 AllocateOutputSetMklShape(context, idx_out, mkl_shape); in ForwardMklTensorInToOutWithMklShape()
[all …]
|