Searched defs:tf_axis (Results 1 – 1 of 1) sorted by relevance
529 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis()2816 for (int tf_axis : squeeze_dims) { in ConvertSqueeze() local4768 const int64 tf_axis = attrs.get<int64>("axis"); in ConvertPack() local4962 const TRT_TensorOrWeights& input, int tf_axis, in ConvertSplitHelper()5030 int tf_axis = inputs.at(0).weights().GetSpan<int>()[0]; in ConvertSplit() local5054 const int tf_axis = attrs.get<int64>("axis"); in ConvertUnpack() local5683 int tf_axis = inputs.at(1).weights().GetSpan<int>()[0]; in ConvertArgMinMax() local