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