Searched refs:input_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
173 std::vector<int64> input_broadcast_dims; in TorchGather() local177 input_broadcast_dims.push_back(i); in TorchGather()181 input_broadcast_dims.push_back(i); in TorchGather()184 input_broadcast_dims.push_back(i + 1); in TorchGather()194 mask, BroadcastInDim(input, sizes, input_broadcast_dims), in TorchGather()