Searched defs:add_scalar (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | functional_modules.py | 58 def add_scalar(self, x: Tensor, y: float) -> Tensor: member in FloatFunctional
129 def add_scalar(self, x: Tensor, y: float) -> Tensor: member in FXFloatFunctional
247 def add_scalar(self, x: Tensor, y: float) -> Tensor: member in QFunctional
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| H A D | fuse_add_to_conv.cc | 41 auto add_scalar = absl::get_if<float>(&add_attr.param); in FuseBiasWithAddAttributes() local
114 auto add_scalar = absl::get_if<float>(&add_attr.param); in FuseAddWithConvolution2D() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/itertools/src/ |
| D | size_hint.rs | 23 pub fn add_scalar(sh: SizeHint, x: usize) -> SizeHint { in add_scalar() function
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/itertools-0.11.0/src/ |
| H A D | size_hint.rs | 25 pub fn add_scalar(sh: SizeHint, x: usize) -> SizeHint { in add_scalar() function
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/ |
| H A D | _numeric_suite.py | 287 def add_scalar(self, x: torch.Tensor, y: float) -> torch.Tensor: member in Shadow
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 570 auto add_scalar = getBinaryOpScalarFusionInfo( in quant_fusion_pattern_and_replacements() local
|
| /aosp_15_r20/external/pytorch/torch/utils/tensorboard/ |
| H A D | writer.py | 342 def add_scalar( member in SummaryWriter
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | BinaryOp.cpp | 387 static Tensor add_scalar( in add_scalar() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit_llga_fuser.py | 294 def add_scalar(x): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 609 TEST_F(VulkanAPITest, add_scalar) { in TEST_F() argument
|