Searched defs:tf_axis (Results 1 – 1 of 1) sorted by relevance
460 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis()2405 for (int tf_axis : squeeze_dims) { in ConvertSqueeze() local4275 const int64 tf_axis = attrs.get<int64>("axis"); in ConvertPack() local4421 const TRT_TensorOrWeights& input, int tf_axis, in ConvertSplitHelper()4489 int tf_axis = inputs.at(0).weights().GetSpan<int>()[0]; in ConvertSplit() local4513 const int tf_axis = attrs.get<int64>("axis"); in ConvertUnpack() local5078 int tf_axis = inputs.at(1).weights().GetSpan<int>()[0]; in ConvertArgMinMax() local