Searched defs:num_dim (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_expanded_weights.py | 419 num_dim, argument
478 def convnet(num_classes, num_dim): argument
501 def convnet(num_classes, num_dim): argument
526 def test_instance_norm_model(self, num_dim, device): argument
527 def instance_norm_model(num_classes, num_dim): argument
552 def test_group_norm_model(self, num_dim, device): argument
553 def group_norm_model(num_classes, num_dim): argument
571 def test_layer_norm_model(self, num_dim, device): argument
572 def layer_norm_model(num_classes, num_dim): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/ |
| H A D | dynamic_enqueue_sparse_expander.cc | 38 int64_t num_dim = in ExpandIndices() local
|
| /aosp_15_r20/external/executorch/backends/xnnpack/runtime/ |
| H A D | XNNExecutor.cpp | 194 size_t num_dim; in resize_outputs() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | utils.h | 203 inline at::Tensor getRequantMultiplierTensor(double requant_multiplier, uint8_t num_dim) { in getRequantMultiplierTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | data_format_ops.cc | 223 const string& dst_format_str, int num_dim, in ComputeDstIndex()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | sparse_ops.cc | 239 DimensionHandle num_dim; in __anone91b66370a02() local
|
| /aosp_15_r20/external/angle/third_party/spirv-tools/src/source/val/ |
| H A D | validate_type.cpp | 715 const auto num_dim = inst->operands().size() - 3; in ValidateTypeTensorViewNV() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor.h | 514 const int num_dim = input_tensor.shape().size(); in Split() local
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_pooling.py | 1501 self, num_dim, adaptive=False, device="cpu", dtype=torch.float argument
|