Searched refs:broadcast_dims_size (Results 1 – 1 of 1) sorted by relevance
64 const int64 broadcast_dims_size = broadcast_dims.size(); in BilinearWeights() local66 std::vector<int64> last_two_dims_indices = {(broadcast_dims_size - 2), in BilinearWeights()67 (broadcast_dims_size - 1)}; in BilinearWeights()104 {broadcast_dims_size - 1}); in BilinearWeights()