Searched defs:index_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
172 std::vector<int64> index_broadcast_dims; in TorchGather() local232 std::vector<int64> index_broadcast_dims; in TorchScatterDense() local