Searched refs:TensorShapeArrayToTrtDims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 247 Status TensorShapeArrayToTrtDims(const Container& shape, nvinfer1::Dims* out, in TensorShapeArrayToTrtDims() function
2670 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims)); in ConvertExpandDims()
2779 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(*input_dims, &new_dims)); in SqueezeTensor()
2880 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(begin, &begin_dims, in ConvertStridedSliceHelper()
2882 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(size, &size_dims, in ConvertStridedSliceHelper()
2884 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(stride, &stride_dims, in ConvertStridedSliceHelper()
2931 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &reshape_dims, in ConvertStridedSliceHelper()
3044 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(input_dims, &new_dims, in ConvertStridedSliceHelper()
4220 TensorShapeArrayToTrtDims(bias_shape_vec, &bias_shape)); in ConvertBiasAdd()
4777 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(tensor_dims, &expanded_dims)); in ConvertPack()
[all …]
|