Searched defs:tf_axis (Results 1 – 1 of 1) sorted by relevance
610 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis()2719 for (int tf_axis : squeeze_dims) { in ConvertSqueeze() local4362 const int64 tf_axis = attrs.get<int64>("axis"); in ConvertPack() local4559 const TRT_TensorOrWeights& input, int tf_axis, in ConvertSplitHelper()4614 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local4626 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local4659 int tf_axis = inputs.at(0).weights().GetSpan<int>()[0]; in ConvertSplit() local4682 const int tf_axis = attrs.get<int64>("axis"); in ConvertUnpack() local5940 int tf_axis = inputs.at(1).weights().GetSpan<int>()[0]; in ConvertArgMinMax() local