Searched refs:ratio_broadcast_indices (Results 1 – 1 of 1) sorted by relevance
76 std::vector<int64> ratio_broadcast_indices(broadcast_dims.size()); in BilinearWeights() local77 std::iota(ratio_broadcast_indices.begin(), ratio_broadcast_indices.end(), 0); in BilinearWeights()78 ratio_broadcast_indices.erase(ratio_broadcast_indices.end() - 2); in BilinearWeights()81 xla::BroadcastInDim(ratio, broadcast_dims, ratio_broadcast_indices); in BilinearWeights()