Searched defs:tf_axis (Results 1 – 1 of 1) sorted by relevance
550 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() local3648 int64_t tf_axis{0}; in ConvertPack() local3899 const TRT_TensorOrWeights& input, int tf_axis, in ConvertSplitHelper()3957 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local3968 const int tf_axis = trt_axis + (params->use_implicit_batch ? 1 : 0); in ConvertSplitHelper() local4015 int tf_axis = inputs.at(0).weights().GetSpan<int>()[0]; in ConvertSplit() local4038 int tf_axis = 0, num = 0; in ConvertUnpack() local4798 int tf_axis = inputs.at(1).weights().GetSpan<int>()[0]; in ConvertArgMinMax() local