Home
last modified time | relevance | path

Searched defs:gather (Results 1 – 25 of 128) sorted by relevance

123456

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_2/
Dgather.example.cpp7 namespace generated_tests::gather { namespace
74 namespace generated_tests::gather { namespace
175 namespace generated_tests::gather { namespace
242 namespace generated_tests::gather { namespace
343 namespace generated_tests::gather { namespace
410 namespace generated_tests::gather { namespace
511 namespace generated_tests::gather { namespace
578 namespace generated_tests::gather { namespace
645 namespace generated_tests::gather { namespace
746 namespace generated_tests::gather { namespace
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc225 xla::XlaOp gather; in XlaGatherWithBatchDimsOpImpl() local
253 xla::XlaOp gather; in Compile() local
297 xla::XlaOp gather; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc168 const HloGatherInstruction* gather, const Shape& output_shape, in PartitionGatherIndexPassthroughPartition()
271 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherPassthroughOperand()
375 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherTrivialIndexedOperandDimension()
486 const HloGatherInstruction* gather, Shape output_shape, in PartitionGatherIndexParallelDimensions()
638 const HloGatherInstruction* gather, PartitionedHlo& operand, in PartitionGather()
686 auto gather = Cast<HloGatherInstruction>(hlo); in HandleGather() local
H A Dspmd_partitioner_test.cc6757 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
6788 auto gather = op::Gather(op::Parameter(0), op::Subtract(clamp, min)); in TEST_F() local
9212 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9242 auto gather = AllOf(op::Shape("s32[2,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9273 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9303 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9334 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9365 auto gather = AllOf(op::Shape("s32[1,4,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9396 auto gather = AllOf(op::Shape("s32[4,1,2,2]"), op::Gather(operand, indices)); in TEST_F() local
9430 auto gather = AllOf(op::Shape("s32[4,2,2,2]"), op::Gather(operand, indices)); in TEST_F() local
[all …]
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/algorithm/include/boost/algorithm/
Dgather.hpp87 std::pair<BidirectionalIterator, BidirectionalIterator> gather in gather() function
111 gather ( in gather() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dgather_simplifier.cc35 auto* gather = DynCast<HloGatherInstruction>(inst); in ExpandInstruction() local
129 auto* gather = DynCast<HloGatherInstruction>(inst); in InstructionMatchesPattern() local
/aosp_15_r20/external/executorch/kernels/optimized/vec/vec256/
H A Dvec256.h77 inline gather(const double* base_addr, const Vectorized<int64_t>& vindex) { in gather() function
83 inline gather(const float* base_addr, const Vectorized<int32_t>& vindex) { in gather() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/vec512/
H A Dvec512.h89 inline gather(const double* base_addr, const Vectorized<int64_t>& vindex) { in gather() function
95 inline gather(const float* base_addr, const Vectorized<int32_t>& vindex) { in gather() function
/aosp_15_r20/packages/modules/Bluetooth/system/blueberry/tests/gd/cert/
Dtest_decorators.py196 def gather(self, *args, **kwargs): member in _TestInfoDecoratorFunc
263 def gather(self, *args, **kwargs): member in _TestInfoBinding
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/vec256/
H A Dvec256.h105 inline gather(const double* base_addr, const Vectorized<int64_t>& vindex) { in gather() function
111 inline gather(const float* base_addr, const Vectorized<int32_t>& vindex) { in gather() function
/aosp_15_r20/tools/test/connectivity/acts/framework/acts/
Dtest_decorators.py219 def gather(self, *args, **kwargs): member in _TestInfoDecoratorFunc
348 def gather(self, *args, **kwargs): member in _TestInfoBinding
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dtensor_array_ops.py296 def gather(self, indices, name=None): member in _GraphTensorArray
568 def gather(self, indices, name=None): member in _GraphTensorArrayV2
865 def gather(self, indices, name=None): member in _EagerTensorArray
1203 def gather(self, indices, name=None): member in TensorArray
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dgather_op_test.py319 def gather(x, indices, axis): function
483 def gather(params): function
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_ops_nccl.py474 def gather(output_t, input_t, rootRank): function
509 def gather(output_t, input_t, rootRank): function
/aosp_15_r20/external/libdav1d/src/arm/32/
H A Dfilmgrain16.S1249 .macro gather dst1, dst2, dst3, dst4, src1, src2, src3, src4, src5, src6, src7, src8 macro
1258 gather d8, d9, d10, d11, d0, d1, d2, d3, d4, d5, d6, d7 define
H A Dfilmgrain.S1386 .macro gather dst1, dst2, dst3, dst4, src1, src2, src3, src4 macro
1395 gather d8, d9, d10, d11, d0, d1, d2, d3 define
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_array_ops_decomposition.cc143 } else if (auto gather = in GetTensorArrayElementShape() local
426 TF::TensorArrayGatherV3Op gather, in HandleTensorArrayGatherV3Op()
889 } else if (auto gather = llvm::dyn_cast<TF::TensorArrayGatherV3Op>(&op)) { in DecomposeTensorArrayOps() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DGather.cpp22 namespace gather { namespace
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc282 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
298 auto gather = Gather(data_operand, indices, dim_numbers, {1}); in TEST_F() local
316 auto gather = Gather(operand1, indices, dim_numbers, {1}); in TEST_F() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DGather.cpp40 SimpleTensor<T> gather(const SimpleTensor<T> &src, const SimpleTensor<uint32_t> &indices, uint32_t … in gather() function
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Dcomm.py218 def gather(tensors, dim=0, destination=None, *, out=None): function
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DGather.cpp27 namespace gather { namespace
/aosp_15_r20/external/python/cpython2/Lib/curses/
Dtextpad.py160 def gather(self): member in Textbox
/aosp_15_r20/external/python/cpython3/Lib/curses/
Dtextpad.py160 def gather(self): member in Textbox
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/python3/lib/python3.11/curses/
Dtextpad.py160 def gather(self): member in Textbox

123456