Searched defs:reshaped_dims (Results 1 – 4 of 4) sorted by relevance
165 dims* reshaped_dims) { in ExpandInputDimsToOutputShape()
458 std::vector<int64> reshaped_dims(warp_dims_without_last_dims.size()); in CalculateGradWarp() local
893 std::vector<int64> reshaped_dims(dot_lhs->shape().dimensions().begin(), in PartitionBaseCase() local916 std::vector<int64> reshaped_dims(dot_rhs->shape().dimensions().begin(), in PartitionBaseCase() local
354 std::vector<int64> reshaped_dims; in RewriteDynamicReshapeSplitInput() local