Searched refs:TensorShapeArrayToTrtDims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 213 Status TensorShapeArrayToTrtDims(const std::vector<int>& shape, in TensorShapeArrayToTrtDims() function
2220 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertExpandDims()
2281 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertSqueeze()
2326 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(begin, &begin_dims, in ConvertStridedSliceHelper()
2328 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(size, &size_dims, in ConvertStridedSliceHelper()
2330 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(stride, &stride_dims, in ConvertStridedSliceHelper()
2370 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &reshape_dims, in ConvertStridedSliceHelper()
2480 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertStridedSliceHelper()
|