Searched defs:broadcasted_dims (Results 1 – 1 of 1) sorted by relevance
198 std::vector<int64> broadcasted_dims, int64 last_warp_dim, in BoundSamples()282 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in CalculateGradData() local583 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local668 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local