Searched defs:output_inds (Results 1 – 3 of 3) sorted by relevance
47 int64* __restrict__ output_inds, T* __restrict__ output_vals, in SparseConcatKernel()89 const int64* __restrict__ permutation, int64* __restrict__ output_inds, in SparseConcatPermuteKernel()243 Tensor* output_inds = nullptr; in allocate_outputs() local
199 Tensor* output_inds = nullptr; in Compute() local
355 Tensor* output_inds = nullptr; in AllocateOutputs() local