Searched refs:GetTfShape (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 101 TensorShape tf_shape0 = input_shape_0.GetTfShape(); in Compute()
102 TensorShape tf_shape1 = input_shape_1.GetTfShape(); in Compute()
213 if (mkl_shape->GetTfShape().num_elements() == in Compute()
|
| D | mkl_concat_op.cc | 184 ? mkl_input_shapes[0].GetTfShape() in Compute()
192 s.IsMklTensor() ? s.GetTfShape() : input_tensors[i].shape(); in Compute()
282 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()
460 tf_input_shapes.push_back(mkl_input_shapes[i].GetTfShape()); in CallEigenVersion()
507 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat()
|
| D | mkl_reshape_op.cc | 72 ? mkl_shape_input.GetTfShape().num_elements() in Compute()
123 TensorShape shape_from = mkl_shape_input.GetTfShape(); in Compute()
|
| D | mkl_conv_ops.h | 285 GetFilterSizeInMklOrder(GetTfShape(context_, src_index), in GetFilterSizeInMklOrder()
286 GetTfShape(context_, filter_index), filter_dims, in GetFilterSizeInMklOrder()
486 auto input_tf_shape = GetTfShape(context_, src_index); in GetOutputAndPadSizeInMklOrder()
487 auto filter_tf_shape = GetTfShape(context_, filter_index); in GetOutputAndPadSizeInMklOrder()
|
| D | mkl_aggregate_ops.cc | 66 src1_shape = input1_in_mkl_format ? src1_mkl_shape.GetTfShape() in Compute()
68 src2_shape = input2_in_mkl_format ? src2_mkl_shape.GetTfShape() in Compute()
|
| D | mkl_tfconv_op.h | 98 TensorShape output_shape = input_shape.GetTfShape(); in ConvertMklToTf()
|
| D | mkl_conv_grad_filter_ops.cc | 398 TensorShape diff_dst_tf_shape = GetTfShape(context, kOutbpropIdx); in Compute()
458 TensorShape obp_tf_shape = GetTfShape(context, 2); in Compute()
623 return GetTfShape(context, input_idx); in MakeInputTfShape()
|
| D | mkl_softmax_op.cc | 60 ? src_mkl_shape.GetTfShape() in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 335 TensorShape diff_dst_tf_shape = GetTfShape(context, kOutbpropIdx); in Compute()
509 return GetTfShape(context, kInputIndex_Filter); in MakeFilterTfShape()
|
| D | mkl_fused_batch_norm_op.cc | 529 tf_shape_src = dnn_shape_src.GetTfShape(); in Compute()
854 tf_shape_diff_dst = dnn_shape_diff_dst.GetTfShape(); in Compute()
868 tf_shape_src = dnn_shape_src.GetTfShape(); in Compute()
|
| D | mkl_slice_op.cc | 86 ? input_mkl_shape.GetTfShape() in ValidateMklInputs()
|
| D | mkl_conv_ops.cc | 962 auto src_tf_shape = GetTfShape(context, kInputIndex_Src); in Compute()
963 auto filter_tf_shape = GetTfShape(context, kInputIndex_Filter); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 39 TensorShape a_mkldnn_tf_shape = a_mkldnn_shape.GetTfShape(); in TEST()
51 TensorShape b_mkldnn_tf_shape = b_mkldnn_shape.GetTfShape(); in TEST()
|
| D | mkl_util.h | 486 return this->GetTfShape() == input_shape.GetTfShape() &&
501 return this->GetTfShape() == input_shape;
588 inline TensorShape GetTfShape() const { in GetTfShape() function
839 TensorShape output_shape = mkl_shape.GetTfShape(); in ConvertMklToTF()
944 inline TensorShape GetTfShape(OpKernelContext* context, size_t input_idx) { in GetTfShape() function
952 return input_mkl_shape.GetTfShape(); in GetTfShape()
|