Searched refs:ConvertAxis (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 529 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis() function
2657 TF_RETURN_IF_ERROR(ConvertAxis(axis[0], dims.nbDims + 1, node_def.name(), in ConvertExpandDims()
2820 TF_RETURN_IF_ERROR(ConvertAxis(tf_axis, dims.nbDims, node_def.name(), in ConvertSqueeze()
4677 ConvertAxis(tf_axes_list[i], tensor->getDimensions().nbDims, in ConvertReduce()
4770 TF_RETURN_IF_ERROR(ConvertAxis(tf_axis, dims.nbDims + 1, node_def.name(), in ConvertPack()
4968 TF_RETURN_IF_ERROR(ConvertAxis(tf_axis, dims.nbDims, node_def.name(), in ConvertSplitHelper()
5136 TF_RETURN_IF_ERROR(ConvertAxis(axis[0], dim.nbDims, node_def.name(), in ConvertConcat()
5334 TF_RETURN_IF_ERROR(ConvertAxis( in ConvertGather()
5686 TF_RETURN_IF_ERROR(ConvertAxis(tf_axis, dims.nbDims, node_def.name(), in ConvertArgMinMax()
|