Searched refs:GetDimensionSizeFromEnd (Results 1 – 2 of 2) sorted by relevance
380 class GetDimensionSizeFromEnd<string dimFromEnd>: NativeCodeCall<382 " GetDimensionSizeFromEnd($0, " # dimFromEnd # "))"415 [(HLO_ConstOp:$m_dim (GetDimensionSizeFromEnd<"1"> $input, $num_lower)),416 (HLO_ConstOp:$n_dim (GetDimensionSizeFromEnd<"0"> $input, $num_upper)),
777 static int GetDimensionSizeFromEnd(Value input, int dim_from_end) { in GetDimensionSizeFromEnd() function792 int dim_0 = GetDimensionSizeFromEnd(input, 1); in Get2DTensorType()793 int dim_1 = GetDimensionSizeFromEnd(input, 0); in Get2DTensorType()