Searched refs:in_dim_size (Results 1 – 1 of 1) sorted by relevance
693 std::vector<int64> in_dim_size(out_dim_size.begin(), out_dim_size.end()); in BroadcastInDim() local695 in_dim_size[broadcast_dimensions[i]] = operand_shape.dimensions(i); in BroadcastInDim()698 ShapeUtil::MakeShape(operand_shape.element_type(), in_dim_size); in BroadcastInDim()