Home
last modified time | relevance | path

Searched defs:is_channels_last_contiguous_2d (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/c10/core/
DSymbolicShapeMeta.cpp115 DEFINE_EAGER_SYMBOOL_COMPUTE(compute_channels_last_contiguous_2d, is_channels_last_contiguous_2d, _… in DEFINE_EAGER_SYMBOOL_COMPUTE() argument
DSymNodeImpl.h135 virtual SymNode is_channels_last_contiguous_2d( in is_channels_last_contiguous_2d() function
/external/pytorch/torch/csrc/utils/
Dpython_symnode.h69 TORCH_SYMNODE_SIZES_STRIDES(is_channels_last_contiguous_2d) in TORCH_SYMNODE_SIZES_STRIDES() argument
/external/pytorch/torch/_prims_common/
D__init__.py253 def is_channels_last_contiguous_2d(a: Tensor) -> bool: function
/external/pytorch/torch/fx/experimental/
Dsym_node.py386 def is_channels_last_contiguous_2d(self, sizes, strides) -> "SymNode": member in SymNode