Searched defs:tmp_indices (Results 1 – 3 of 3) sorted by relevance
438 auto tmp_indices = at::empty(self.sizes(), self.options().dtype(kLong)); in kthvalue_out_impl_cpu() local462 TensorAccessor<int64_t, 1> tmp_indices( in kthvalue_out_impl_cpu() local
734 Tensor tmp_indices; in Compute() local
605 void *tmp_indices = all_prims_supported ? NULL : malloc(max_index_count * sizeof(uint32_t)); in compile_vertex_list() local