Searched defs:expanded_dims (Results 1 – 2 of 2) sorted by relevance
475 HloInstruction* operand, absl::Span<const int64> expanded_dims) { in ExpandFirstDimIntoNDims()
4776 nvinfer1::Dims expanded_dims; in ConvertPack() local