Searched refs:to_dim (Results 1 – 2 of 2) sorted by relevance
386 for (int64_t to_dim = common_factors[i + 1].second - 1; in ConvertDimensionNumbers() local387 to_dim >= common_factors[i].second; --to_dim) { in ConvertDimensionNumbers()388 const int64_t to_size = to_sizes[to_dim]; in ConvertDimensionNumbers()392 dimensions.to_dimensions.push_back(to_dim); in ConvertDimensionNumbers()
719 int64_t to_dim = broadcast_dimensions[from_dim]; in BinaryOp() local720 to_size[to_dim] = from_shape.dimensions(from_dim); in BinaryOp()721 to_size_is_dynamic[to_dim] = from_shape.is_dynamic_dimension(from_dim); in BinaryOp()