| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | instancenorm.h | 11 inline Tensor instance_norm( in instance_norm() function
20 return torch::instance_norm( in instance_norm()
35 /// https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.instance_norm
44 /// F::instance_norm(input,
47 inline Tensor instance_norm(
50 return detail::instance_norm(
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | modules.cpp | 1685 InstanceNorm1d instance_norm( in TEST_F() local
1688 ASSERT_TRUE(instance_norm->options.track_running_stats()); in TEST_F()
1690 ASSERT_TRUE(instance_norm->running_mean.defined()); in TEST_F()
1691 ASSERT_EQ(instance_norm->running_mean.dim(), 1); in TEST_F()
1692 ASSERT_EQ(instance_norm->running_mean.size(0), 5); in TEST_F()
1694 ASSERT_TRUE(instance_norm->running_var.defined()); in TEST_F()
1695 ASSERT_EQ(instance_norm->running_var.dim(), 1); in TEST_F()
1696 ASSERT_EQ(instance_norm->running_var.size(0), 5); in TEST_F()
1698 ASSERT_TRUE(instance_norm->num_batches_tracked.defined()); in TEST_F()
1699 ASSERT_EQ(instance_norm->num_batches_tracked.dim(), 0); in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
| H A D | instance_norm_expanded_weights.py | 17 @implements_per_sample_grads(F.instance_norm)
21 instance_norm = partial(torch.instance_norm, cudnn_enabled=True)
25 output = forward_helper(instance_norm, expanded_args, expanded_kwargs)
94 "ni...->ni", F.instance_norm(input, eps=eps) * grad_output
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | decomposition_skip.py | 164 op_callable = torch.instance_norm # type: ignore[attr-defined]
166 new_op_name = "instance_norm"
181 torch.instance_norm = torch.ops.onnx_export.instance_norm # type: ignore[attr-defined]
193 torch.instance_norm = cls.op_callable # type: ignore[attr-defined]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_instance_norm_test.cc | 90 auto instance_norm = QuantizedInstanceNorm( in Expect() local
101 {instance_norm.y, instance_norm.y_min, instance_norm.y_max}, in Expect()
|
| /aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
| H A D | normalization.py | 58 y = self.instance_norm(self.data)
63 return "instance_norm"
66 return tensor_engine.is_supported(self.instance_norm)
|
| H A D | pt_engine.py | 36 def instance_norm(self, data): member in TorchTensorEngine
37 return torch.nn.functional.instance_norm(data)
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | mappings.py | 211 # F.instance_norm
213 F.instance_norm,
496 F.instance_norm,
524 toq.instance_norm,
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | normalization.py | 181 return torch.ops.quantized.instance_norm(
248 return torch.ops.quantized.instance_norm(
315 return torch.ops.quantized.instance_norm(
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 329 ("nn.functional.instance_norm", 124),
436 ("nn.InstanceNorm2d", 16426, "nn.functional.instance_norm"),
467 ("nn.InstanceNorm1d", 3658, "nn.functional.instance_norm"),
473 ("nn.InstanceNorm3d", 2893, "nn.functional.instance_norm"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | instancenorm.h | 62 /// Options for `torch::nn::functional::instance_norm`.
67 /// F::instance_norm(input,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/compatibility/ |
| H A D | tf_upgrade_v2_safety_test.py | 162 "tf.contrib.layers.instance_norm":
189 "tf.contrib.layers.instance_norm":
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | quantization_mappings.py | 184 F.instance_norm: torch.ops.quantized.instance_norm,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | check_alias_annotation.cpp | 226 // batch_norm and instance_norm have incorrect annotations, because
234 case aten::instance_norm: in shouldIgnoreNode()
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 301 - aten::instance_norm
689 - quantized::instance_norm
1038 quantized::instance_norm: 1
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | common_utils.py | 225 "nn.functional.instance_norm",
232 op_name == "nn.functional.instance_norm"
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 927 auto instance_norm = getObservedQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
928 "aten::instance_norm", in quant_fusion_pattern_and_replacements()
929 "quantized::instance_norm", in quant_fusion_pattern_and_replacements()
1045 std::move(instance_norm), in quant_fusion_pattern_and_replacements()
|
| H A D | helper.cpp | 31 "instance_norm",
51 "instance_norm",
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | instancenorm_test.py | 29 return F.instance_norm(input, weight=weight, bias=bias, eps=eps)
|
| H A D | qinstancenorm_test.py | 37 return torch.ops.quantized.instance_norm(
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _decomp.py | 54 aten.instance_norm.default,
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_fx_to_onnx_decomp_skip.py | 67 return torch.nn.functional.instance_norm(x)
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_alias_analysis.cpp | 613 std::make_tuple("aten::instance_norm", aten::instance_norm, false),
615 std::make_tuple("aten::instance_norm", aten::instance_norm, true)));
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | nn.functional.rst | 108 instance_norm
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 479 nn.functional.instance_norm
|