Home
last modified time | relevance | path

Searched defs:memory_format (Results 1 – 25 of 125) sorted by relevance

12345

/external/executorch/kernels/test/
Dop_empty_test.cpp31 optional<MemoryFormat> memory_format, in op_empty_out()
42 optional<MemoryFormat> memory_format; in test_empty_out() local
63 optional<MemoryFormat> memory_format; in ET_FORALL_REAL_TYPES_AND() local
74 optional<MemoryFormat> memory_format; in TEST_F() local
88 optional<MemoryFormat> memory_format; in TEST_F() local
Dop_full_like_test.cpp32 optional<MemoryFormat> memory_format, in op_full_like_out()
45 MemoryFormat memory_format = MemoryFormat::Contiguous; in test_full_like_out() local
63 MemoryFormat memory_format; in test_full_like_out_mismatched_shape() local
77 MemoryFormat memory_format = MemoryFormat::Contiguous; in test_full_like_out() local
193 optional<MemoryFormat> memory_format; in TEST_F() local
/external/pytorch/torch/distributed/_shard/sharded_tensor/
D__init__.py33 memory_format=torch.contiguous_format, argument
88 memory_format=torch.contiguous_format, argument
142 memory_format=torch.contiguous_format, argument
198 memory_format=torch.contiguous_format, argument
252 memory_format=torch.contiguous_format, argument
308 memory_format=torch.contiguous_format, argument
/external/pytorch/torch/nn/utils/
Dmemory_format.py5 def convert_conv2d_weight_memory_format(module, memory_format): argument
80 def convert_conv3d_weight_memory_format(module, memory_format): argument
/external/pytorch/torch/csrc/lazy/ts_backend/ops/
Dto_copy.h26 const std::optional<at::MemoryFormat>& memory_format, in ToCopy()
55 const std::optional<at::MemoryFormat>& memory_format) const { in CanBeReused()
123 std::optional<at::MemoryFormat> memory_format; variable
/external/pytorch/torch/csrc/utils/
Dtensor_memoryformats.cpp19 PyObject* getTHPMemoryFormat(at::MemoryFormat memory_format) { in getTHPMemoryFormat()
36 PyObject* memory_format = THPMemoryFormat_New(format, module_name + name); in initializeMemoryFormats() local
/external/pytorch/aten/src/ATen/native/utils/
DFactory.cpp14 const c10::MemoryFormat memory_format, in empty_with_tail_padding()
38 const c10::MemoryFormat memory_format) { in allocate_padded_contiguous_if_needed()
/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorFactories.cpp23 auto memory_format = in verify_empty_parameters() local
47 auto memory_format = options.memory_format_opt().value_or(MemoryFormat::Preserve); in empty_like_nested() local
107 auto memory_format = optional_memory_format.value_or(MemoryFormat::Preserve); in _to_copy_nested() local
135 auto memory_format = optional_memory_format.value_or(c10::MemoryFormat::Preserve); in clone_nested() local
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DFactory.cpp17 const std::optional<MemoryFormat> memory_format) { in _empty_affine_quantized()
37 const std::optional<MemoryFormat> memory_format) { in empty_memory_format()
/external/pytorch/aten/src/ATen/cudnn/
DDescriptors.cpp36 void TensorDescriptor::set(const at::Tensor &t, at::MemoryFormat memory_format, size_t pad) { in set()
43 auto memory_format = t.suggest_memory_format(); in set() local
124 void FilterDescriptor::set(const at::Tensor &t, const at::MemoryFormat memory_format, int64_t pad) { in set()
/external/pytorch/torch/onnx/
Dsymbolic_opset8.py356 memory_format=None, argument
370 memory_format=None, argument
391 memory_format=None, argument
412 memory_format=None, argument
441 memory_format=None, argument
/external/pytorch/aten/src/ATen/core/
DCheckMemoryFormat.h8 std::optional<MemoryFormat> memory_format) { in check_tensor_options_and_extract_memory_format()
/external/pytorch/aten/src/ATen/native/mkldnn/
DConv.cpp158 auto memory_format = at::MemoryFormat::Contiguous; in mkldnn_convolution_memory_format() local
177 auto memory_format = mkldnn_convolution_memory_format(input_t.ndimension(), is_channels_last); in _mkldnn_convolution_out() local
246 auto memory_format = in _mkldnn_convolution() local
401 auto memory_format = in mkldnn_convolution_pointwise_binary() local
647 auto memory_format = in _mkldnn_convolution_transpose() local
787 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_backward_input() local
858 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_backward_weights() local
870 auto memory_format = mkldnn_convolution_memory_format(input_t.ndimension(), is_channels_last); in mkldnn_convolution_backward() local
937 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_transpose_backward_input() local
1009 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_transpose_backward_weights() local
[all …]
/external/pytorch/aten/src/ATen/native/
DAveragePool2d.cpp54 auto memory_format = input.suggest_memory_format(); in TORCH_PRECOMPUTE_META_FUNC() local
133 auto memory_format = input.suggest_memory_format(); in TORCH_META_FUNC() local
Dgroup_norm.cpp80 auto memory_format = X.device().is_cpu() ? in native_group_norm() local
127 auto memory_format = X.device().is_cpu() ? in native_group_norm_backward() local
199 auto memory_format = input.suggest_memory_format(); in group_norm() local
DDilatedMaxPool2d.cpp49 const auto memory_format = input.suggest_memory_format(); in TORCH_META_FUNC() local
125 const auto memory_format = input.suggest_memory_format(); in TORCH_META_FUNC() local
DPixelShuffle.cpp34 auto memory_format = self.suggest_memory_format(); in pixel_shuffle_cpu() local
62 auto memory_format = self.suggest_memory_format(); in pixel_unshuffle_cpu() local
/external/pytorch/aten/src/ATen/native/cudnn/
DConvShared.cpp113 at::MemoryFormat memory_format) { in setConvolutionParams()
220 auto memory_format = output->suggest_memory_format(); in cudnn_convolution_forward_out() local
252 auto memory_format = cudnn_conv_suggest_memory_format(input_t, weight_t); in cudnn_convolution() local
321 auto memory_format = in cudnn_convolution_transpose_backward_input() local
380 auto memory_format = cudnn_conv_suggest_memory_format(*grad_output, *weight); in cudnn_convolution_backward_input() local
713 auto memory_format = cudnn_conv_suggest_memory_format(input_t, weight_t); in cudnn_convolution_relu() local
767 auto memory_format = cudnn_conv_suggest_memory_format(input_t, weight_t); in cudnn_convolution_add_relu() local
/external/executorch/kernels/portable/cpu/
Dop_empty.cpp29 exec_aten::optional<exec_aten::MemoryFormat> memory_format, in empty_out()
Dop_clone.cpp24 exec_aten::optional<exec_aten::MemoryFormat> memory_format, in clone_out()
/external/pytorch/torch/csrc/
DMemoryFormat.h12 PyObject_HEAD at::MemoryFormat memory_format; member
/external/pytorch/torch/csrc/jit/runtime/static/
Dops.h110 c10::MemoryFormat memory_format) { in create_empty_from()
123 c10::MemoryFormat memory_format) { in create_empty_from()
/external/pytorch/test/nn/
Dtest_dropout.py95 def _test_dropout(self, cls, device, input, memory_format=torch.contiguous_format): argument
127 self, cls, device, memory_format=torch.contiguous_format argument
/external/executorch/exir/
Ddim_order_utils.py53 memory_format: Optional[torch.memory_format], ndim: int argument
/external/pytorch/torch/_inductor/
Ddecomposition.py514 memory_format: Optional[torch.memory_format] = None, argument
529 memory_format: Optional[torch.memory_format] = None, argument
546 memory_format: Optional[torch.memory_format] = None, argument
567 memory_format: torch.memory_format = torch.preserve_format, argument
586 memory_format: Optional[torch.memory_format] = None, argument
607 memory_format: Optional[torch.memory_format] = None, argument

12345