Searched refs:create_empty_from (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | ops.h | 54 inline at::Tensor create_empty_from(const at::Tensor& t) { in create_empty_from() function
64 inline at::Tensor create_empty_from( in create_empty_from() function
81 inline at::Tensor create_empty_from( in create_empty_from() function
88 inline at::Tensor create_empty_from(const at::Tensor& t, c10::Layout layout) { in create_empty_from() function
98 inline at::Tensor create_empty_from(const at::Tensor& t, c10::Device device) { in create_empty_from() function
108 inline at::Tensor create_empty_from( in create_empty_from() function
120 inline at::Tensor create_empty_from( in create_empty_from() function
|
| D | ops.cpp | 650 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b1402()
693 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b1602()
717 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b1602()
740 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b1902()
914 p_node->Output(0) = create_empty_from(p_node->Input(0).toTensor()); in __anon5bf8650b2102()
947 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b2502()
970 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b2702()
1031 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b2b02()
1063 p_node->Output(0) = create_empty_from(in0_t); in __anon5bf8650b2d02()
1111 p_node->Output(0) = create_empty_from(src, memory_format); in __anon5bf8650b2f02()
[all …]
|
| /external/pytorch/aten/src/ATen/native/quantized/cuda/ |
| D | EmbeddingBag.cu | 166 inline at::Tensor create_empty_from( in create_empty_from() function
239 sample_weights = create_empty_from(output, kFloat); in embedding_bag_byte_impl()
278 auto output = create_empty_from(weight, at::kFloat); in embedding_bag_byte_rowwise_offsets()
421 sample_weights = create_empty_from(output, kFloat); in embedding_bag_4bit_impl()
459 auto output = create_empty_from(weight, at::kFloat); in embedding_bag_4bit_rowwise_offsets()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qembeddingbag.cpp | 914 inline at::Tensor create_empty_from( in create_empty_from() function
931 auto output = create_empty_from(weight, at::kFloat); in embedding_bag_byte_rowwise_offsets()
956 auto output = create_empty_from(weight, at::kFloat); in embedding_bag_4bit_rowwise_offsets()
981 auto output = create_empty_from(weight, at::kFloat); in embedding_bag_2bit_rowwise_offsets()
|