Searched defs:offsets_data (Results 1 – 5 of 5) sorted by relevance
174 const auto* offsets_data = offsets.const_data_ptr<index_t>(); in _segment_reduce_offsets_cpu_kernel() local367 const auto* offsets_data = offsets_contig.const_data_ptr<index_t>(); in _segment_reduce_cpu_offsets_backward_kernel() local
209 auto* offsets_data = offsets.const_data_ptr<index_t>(); in index_select_add() local387 auto* offsets_data = offsets.const_data_ptr<index_t>(); in index_select_add() local575 auto* offsets_data = offsets.const_data_ptr<index_t>(); in index_select_scale_add() local764 auto* offsets_data = offsets.const_data_ptr<index_t>(); in index_select_scale_add() local
237 auto offsets_data = offsets.data_ptr<OffsetType>(); in embedding_bag_nbit_impl() local409 auto offsets_data = offsets.data_ptr<OffsetType>(); in embedding_bag_byte_impl() local
184 std::vector<int32_t> offsets_data(dst_depth * 2 * 4, 0); in CreateConvPointwise() local
313 auto* offsets_data = offsets.const_data_ptr<index_t>(); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local