Home
last modified time | relevance | path

Searched refs:PrepareTensorForShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1702 Status PrepareTensorForShape(Converter* converter, in PrepareTensorForShape() function
2627 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertReshape()
2671 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertExpandDims()
2780 TF_RETURN_IF_ERROR(PrepareTensorForShape( in SqueezeTensor()
2895 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertStridedSliceHelper()
2999 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertStridedSliceHelper()
3023 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertStridedSliceHelper()
3046 TF_RETURN_IF_ERROR(PrepareTensorForShape( in ConvertStridedSliceHelper()
4232 TF_RETURN_IF_ERROR(PrepareTensorForShape(params->converter, inputs.at(0), in ConvertBiasAdd()
4240 TF_RETURN_IF_ERROR(PrepareTensorForShape(params->converter, inputs.at(1), in ConvertBiasAdd()
[all …]
Dconvert_nodes.h697 Status PrepareTensorForShape(Converter* converter,
Dconvert_nodes_test.cc955 PrepareTensorForShape(converter, input, GetTestDims(reshape_dims), in TestPrepareTensorForShape()
971 TEST_F(ConverterTest, PrepareTensorForShape) { in TEST_F() argument