Searched defs:offsets_in (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qembeddingbag.cpp | 561 const std::optional<at::Tensor>& offsets_in, in embedding_bag_byte_helper()
659 const std::optional<at::Tensor>& offsets_in, in _embedding_bag_nbit_helper()
763 const std::optional<at::Tensor>& offsets_in, in embeddingbag_byte()
784 const std::optional<at::Tensor>& offsets_in, in embeddingbag_4bit()
822 const std::optional<Tensor>& offsets_in, in embedding_bag_byte_rowwise_offsets_out()
845 const std::optional<Tensor>& offsets_in, in embedding_bag_4bit_rowwise_offsets_out()
880 const std::optional<Tensor>& offsets_in, in embedding_bag_2bit_rowwise_offsets_out()
924 const std::optional<Tensor>& offsets_in, in embedding_bag_byte_rowwise_offsets()
949 const std::optional<Tensor>& offsets_in, in embedding_bag_4bit_rowwise_offsets()
974 const std::optional<Tensor>& offsets_in, in embedding_bag_2bit_rowwise_offsets()
[all …]
|
| /external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| D | EmbeddingBag.cu | 270 const std::optional<Tensor>& offsets_in, in embedding_bag_byte_rowwise_offsets()
452 const std::optional<Tensor>& offsets_in, in embedding_bag_4bit_rowwise_offsets()
|
| /external/mesa3d/src/gallium/frontends/dri/ |
| D | loader_dri3_helper.c | 1820 uint32_t *strides_in, *offsets_in; in loader_dri3_create_image_from_buffers() local
|