Home
last modified time | relevance | path

Searched defs:fill_value (Results 1 – 25 of 37) sorted by relevance

12

/external/ComputeLibrary/src/core/CPP/kernels/
DCPPUpsampleKernel.cpp89 const uint8_t fill_value = _output->info()->quantization_info().uniform().offset; in run() local
95 const int8_t fill_value = _output->info()->quantization_info().uniform().offset; in run() local
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dpadding.h30 std::optional<double> fill_value; in pad() local
/external/openscreen/platform/base/
Dudp_packet.cc17 UdpPacket::UdpPacket(size_type size, uint8_t fill_value) in UdpPacket()
/external/executorch/backends/cadence/reference/operators/
Dop_full.cpp22 const Scalar& fill_value, in full_out()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/
Dall.cpp97 …const float16_t fill_value = (pool_info.pool_type == PoolingType::MAX) ? fp16_min :… in pooling3_fp16_neon_nchw() local
207 … const T fill_value = (pool_info.pool_type == PoolingType::MAX) ? float_min : 0.f; in pooling2_nchw_maxpool_indices() local
263 const float16_t fill_value = (pool_info.pool_type == PoolingType::MAX) ? fp16_min : 0.0f; in pooling2_fp16_neon_nchw() local
337 const float16_t fill_value = (pool_info.pool_type == PoolingType::MAX) ? fp16_min : 0.0f; in poolingMxN_fp16_neon_nchw() local
424 …const float fill_value = (pool_info.pool_type == PoolingType::MAX) ? -std::numeric_limits<float… in poolingMxN_fp32_neon_nchw() local
513 …const float fill_value = (pool_info.pool_type == PoolingType::MAX) ? -std::numeric_limits<float… in pooling2_fp32_neon_nchw() local
587 …const float fill_value = (pool_info.pool_type == PoolingType::MAX) ? -std::numeric_limits<float… in pooling3_fp32_neon_nchw() local
668 …const float fill_value = (pool_info.pool_type == PoolingType::MAX) ? -std::numeric_limits<float… in pooling7_fp32_neon_nchw() local
/external/executorch/kernels/portable/cpu/
Dop_full.cpp22 const Scalar& fill_value, in full_out()
Dop_full_like.cpp22 const Scalar& fill_value, in full_like_out()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunfold_large_splat_constant.cc87 mlir::arith::ConstantOp fill_value = in MaybeUnfoldLargeSplatConstant() local
/external/XNNPACK/test/
Dfill-microkernel-tester.h82 uint32_t fill_value = 0; in Test() local
Dpad-microkernel-tester.h118 uint32_t fill_value = 0; in Test() local
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DFull.cpp37 const ValueRef fill_value, in add_full_node()
DPad.cpp61 const ValueRef& fill_value, in add_constant_pad_nd_node()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/
Dquantized.h424 …const T fill_value = (pool_info.pool_type == PoolingType::MAX) ? std::numeric_limits<T>::min() : T… in pooling2_quantized_neon_nchw() local
560 …const T fill_value = (pool_info.pool_type == PoolingType::AVG) ? T(0) : std::numeric_limits<T>::… in pooling3_quantized_neon_nchw() local
716 …const T fill_value = (pool_info.pool_type == PoolingType::AVG) ? T(0)… in poolingMxN_quantized_neon_nchw() local
/external/executorch/kernels/test/
Dop_full_test.cpp31 op_full_out(const IntArrayRef sizes, const Scalar& fill_value, Tensor& out) { in op_full_out()
Dop_full_like_test.cpp31 const Scalar& fill_value, in op_full_like_out()
/external/executorch/extension/tensor/
Dtensor_ptr_maker.cpp119 exec_aten::Scalar fill_value, in full_strided()
/external/tensorflow/tensorflow/core/kernels/image/
Dimage_ops.cc88 T fill_value(0); in DoImageProjectiveTransformOp() local
/external/pytorch/aten/src/ATen/native/
DFill.cpp96 Tensor& fill_diagonal_(Tensor& self, const Scalar& fill_value, bool wrap) { in fill_diagonal_()
DTensorFactories.cpp585 const Scalar& fill_value, in infer_full_options()
608 Tensor full(IntArrayRef size, const Scalar& fill_value, in full()
623 Tensor& full_out(IntArrayRef size, const Scalar& fill_value, Tensor& result) { in full_out()
633 const Scalar& fill_value, in full_like()
649 const Scalar& fill_value, in new_full()
1769 const Scalar& fill_value, in full()
/external/pytorch/test/
Dtest_masked.py323 def test_where(self, sparse_kind, fill_value): argument
/external/pytorch/torch/distributed/_shard/sharded_tensor/
D__init__.py192 fill_value, argument
/external/pytorch/torch/_inductor/fx_passes/
Dpost_grad.py441 def pointless_cumsum_replacement(match: Match, shape, fill_value, device, dtype, dim): argument
697 def constant_pad_nd(x, padding, fill_value=0): argument
/external/pytorch/torch/_inductor/
Dlowering.py2724 def _full(fill_value, device, dtype, size): argument
2755 def full_like(x, fill_value, **kwargs): argument
2759 def tensor_constructor(fill_value): argument
2829 def constant_like(fill_value): argument
2838 def new_constant(fill_value): argument
2916 def full(size, fill_value, **kwargs): argument
3652 def constant_pad_nd(x, padding, fill_value=0): argument
3723 x, fill_value, padding=None, pad_fill_value=1.0, dim=None argument
5413 def fill_(x, fill_value): argument
/external/pytorch/torch/onnx/
Dsymbolic_opset8.py436 fill_value, argument
/external/pytorch/torch/distributed/tensor/
D_api.py1060 fill_value, argument

12