Home
last modified time | relevance | path

Searched defs:tf_axis (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc550 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis()
2486 for (int tf_axis : squeeze_dims) { in ConvertSqueeze() local
3648 int64_t tf_axis{0}; in ConvertPack() local
3899 const TRT_TensorOrWeights& input, int tf_axis, in ConvertSplitHelper()
3957 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local
3968 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local
4015 int tf_axis = inputs.at(0).weights().GetSpan<int>()[0]; in ConvertSplit() local
4038 int tf_axis = 0, num = 0; in ConvertUnpack() local
4798 int tf_axis = inputs.at(1).weights().GetSpan<int>()[0]; in ConvertArgMinMax() local