Home
last modified time | relevance | path

Searched refs:reducer (Results 1 – 25 of 130) sorted by relevance

123456

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorReductionCuda.h25 __device__ EIGEN_ALWAYS_INLINE void atomicReduce(T* output, T accum, R& reducer) { in atomicReduce() argument
31 reducer.reduce(accum, reinterpret_cast<T*>(&newval)); in atomicReduce()
39 reducer.reduce(accum, reinterpret_cast<T*>(&newval)); in atomicReduce()
48 reducer.reduce(accum, reinterpret_cast<T*>(&newval)); in atomicReduce()
56 reducer.reduce(accum, reinterpret_cast<T*>(&newval)); in atomicReduce()
84 __device__ inline void atomicReduce(half2* output, half2 accum, R<half>& reducer) { in atomicReduce() argument
87 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval)); in atomicReduce()
95 reducer.reducePacket(accum, reinterpret_cast<half2*>(&newval)); in atomicReduce()
125 __global__ void FullReductionKernel(Reducer reducer, const Self input, Index num_coeffs, in FullReductionKernel() argument
132 *output = reducer.initialize(); in FullReductionKernel()
[all …]
DTensorReduction.h131 …NLINE void reduce(const Self& self, typename Self::Index firstIndex, Op& reducer, typename Self::C…
135 GenericDimReducer<DimIndex-1, Self, Op>::reduce(self, input, reducer, accum);
141 …NLINE void reduce(const Self& self, typename Self::Index firstIndex, Op& reducer, typename Self::C…
144 reducer.reduce(self.m_impl.coeff(input), accum);
150 …ONG_INLINE void reduce(const Self& self, typename Self::Index index, Op& reducer, typename Self::C…
151 reducer.reduce(self.m_impl.coeff(index), accum);
157 …Self& self, typename Self::Index firstIndex, typename Self::Index numValuesToReduce, Op& reducer) {
158 typename Self::CoeffReturnType accum = reducer.initialize();
160 reducer.reduce(self.m_impl.coeff(firstIndex + j), &accum);
162 return reducer.finalize(accum);
[all …]
/external/swiftshader/third_party/SPIRV-Tools/test/reduce/
Dvalidation_during_reduction_test.cpp212 Reducer reducer(env); in TEST() local
213 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
216 reducer.SetInterestingnessFunction( in TEST()
219 reducer.AddReductionPass( in TEST()
233 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
427 Reducer reducer(env); in TEST() local
428 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
431 reducer.SetInterestingnessFunction( in TEST()
434 reducer.AddReductionPass( in TEST()
448 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
[all …]
Dreducer_test.cpp194 Reducer reducer(kEnv); in TEST() local
196 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
197 reducer.SetInterestingnessFunction( in TEST()
201 reducer.AddReductionPass( in TEST()
203 reducer.AddReductionPass( in TEST()
216 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
514 Reducer reducer(kEnv); in TEST() local
516 reducer.SetInterestingnessFunction(InterestingWhileIMulReachable); in TEST()
517 reducer.AddDefaultReductionPasses(); in TEST()
518 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
[all …]
/external/deqp-deps/SPIRV-Tools/test/reduce/
Dvalidation_during_reduction_test.cpp212 Reducer reducer(env); in TEST() local
213 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
216 reducer.SetInterestingnessFunction( in TEST()
219 reducer.AddReductionPass( in TEST()
233 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
427 Reducer reducer(env); in TEST() local
428 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
431 reducer.SetInterestingnessFunction( in TEST()
434 reducer.AddReductionPass( in TEST()
448 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
[all …]
Dreducer_test.cpp194 Reducer reducer(kEnv); in TEST() local
196 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
197 reducer.SetInterestingnessFunction( in TEST()
201 reducer.AddReductionPass( in TEST()
203 reducer.AddReductionPass( in TEST()
216 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
514 Reducer reducer(kEnv); in TEST() local
516 reducer.SetInterestingnessFunction(InterestingWhileIMulReachable); in TEST()
517 reducer.AddDefaultReductionPasses(); in TEST()
518 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/test/reduce/
Dvalidation_during_reduction_test.cpp212 Reducer reducer(env); in TEST() local
213 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
216 reducer.SetInterestingnessFunction( in TEST()
219 reducer.AddReductionPass( in TEST()
233 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
427 Reducer reducer(env); in TEST() local
428 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
431 reducer.SetInterestingnessFunction( in TEST()
434 reducer.AddReductionPass( in TEST()
448 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
[all …]
Dreducer_test.cpp194 Reducer reducer(kEnv); in TEST() local
196 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
197 reducer.SetInterestingnessFunction( in TEST()
201 reducer.AddReductionPass( in TEST()
203 reducer.AddReductionPass( in TEST()
216 Reducer::ReductionResultStatus status = reducer.Run( in TEST()
514 Reducer reducer(kEnv); in TEST() local
516 reducer.SetInterestingnessFunction(InterestingWhileIMulReachable); in TEST()
517 reducer.AddDefaultReductionPasses(); in TEST()
518 reducer.SetMessageConsumer(kMessageConsumer); in TEST()
[all …]
/external/skqp/tests/
DPathOpsCubicReduceOrderTest.cpp63 SkReduceOrder reducer; in DEF_TEST() local
112 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
123 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
126 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
135 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
146 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
157 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
168 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
171 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
180 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
[all …]
DPathOpsQuadReduceOrderTest.cpp25 SkReduceOrder reducer; in oneOffTest() local
26 SkDEBUGCODE(int result = ) reducer.reduce(quad); in oneOffTest()
33 SkReduceOrder reducer; in standardTestCases() local
55 order = reducer.reduce(quad); in standardTestCases()
64 order = reducer.reduce(quad); in standardTestCases()
/external/skia/tests/
DPathOpsCubicReduceOrderTest.cpp63 SkReduceOrder reducer; in DEF_TEST() local
112 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
123 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
126 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
135 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
146 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
157 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
168 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
171 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
180 order = reducer.reduce(cubic, SkReduceOrder::kAllow_Quadratics); in DEF_TEST()
[all …]
DPathOpsQuadReduceOrderTest.cpp25 SkReduceOrder reducer; in oneOffTest() local
26 SkDEBUGCODE(int result = ) reducer.reduce(quad); in oneOffTest()
33 SkReduceOrder reducer; in standardTestCases() local
55 order = reducer.reduce(quad); in standardTestCases()
64 order = reducer.reduce(quad); in standardTestCases()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py41 reducer = grouping.Reducer(
47 grouping.group_by_reducer(lambda x: x % 2, reducer))
60 reducer = grouping.Reducer(
67 lambda x: math_ops.cast(x, dtypes.int64) % 2, reducer))
76 reducer = grouping.Reducer(
84 grouping.group_by_reducer(lambda x, y: y % 2, reducer))
98 reducer = grouping.Reducer(
104 grouping.group_by_reducer(lambda x: x.values[0] % 2, reducer))
120 reducer = grouping.Reducer(
127 grouping.group_by_reducer(lambda x: x, reducer))
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops.h55 const ReductionAxes& reduction_axes, const Reducer& reducer) {
56 out.device(d) = in.reduce(reduction_axes, reducer);
69 const Reducer<ScalarType>& reducer) { \
89 const functor::MeanReducer<Scalar>& reducer) {
106 const functor::MeanReducer<ScalarType>& reducer) { \
133 const functor::EuclideanNormReducer<Scalar>& reducer) {
147 const functor::EuclideanNormReducer<bfloat16>& reducer) {
161 static auto identity(const Reducer& reducer)
162 -> decltype(reducer.initialize()) {
163 return reducer.initialize();
[all …]
Dreduction_ops_common.h172 Reducer reducer;
180 constants.kZero, reducer);
196 Functor::FillIdentity(d, tmp_out.flat<T>(), reducer);
200 constants.kZero, reducer);
204 constants.kZero, reducer);
208 constants.kOne, reducer);
213 constants.kZeroTwo, reducer);
217 constants.kOne, reducer);
235 constants.kOne, reducer);
260 const Reducer& reducer) {
[all …]
/external/llvm-project/mlir/include/mlir/Reducer/
DReductionTreePass.h53 Reducer reducer; in runOnOperation() local
54 std::vector<bool> transformSpace = reducer.initTransformSpace(module); in runOnOperation()
62 LLVM_DEBUG(llvm::dbgs() << "\nReduction Tree Pass: " << reducer.getName();); in runOnOperation()
82 Reducer reducer; variable
103 reducer.generateVariants(currNode, test, 1); in singlePathTraversal()
117 reducer.generateVariants(currNode, test, 2); in singlePathTraversal()
136 reducer.generateVariants(currNode, test, 1); in singlePathTraversal()
/external/rust/crates/rayon/src/iter/collect/
Dtest.rs143 let reducer = consumer.to_reducer(); in left_produces_fewer_items() localVariable
151 reducer.reduce(left_result, right_result) in left_produces_fewer_items()
200 let reducer = consumer.to_reducer(); in reducer_does_not_preserve_order() localVariable
208 reducer.reduce(right_result, left_result) in reducer_does_not_preserve_order()
220 let reducer = consumer.to_reducer(); in right_produces_fewer_items() localVariable
228 reducer.reduce(left_result, right_result) in right_produces_fewer_items()
240 let reducer = consumer.to_reducer(); in left_panics() localVariable
254 reducer.reduce(left_result, right_result) in left_panics()
267 let reducer = consumer.to_reducer(); in right_panics() localVariable
281 reducer.reduce(left_result, right_result) in right_panics()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc86 XlaCompiler::CompilationResult reducer; in Compile() local
89 {reducer_arg, reducer_arg}, &reducer)); in Compile()
96 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
100 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
119 context->Input(0), context->Input(1), *reducer.computation, in Compile()
Dxla_reduce_op.cc82 XlaCompiler::CompilationResult reducer; in Compile() local
87 std::vector<XlaCompiler::Argument>(n_ * 2, reducer_arg), &reducer)); in Compile()
98 xla::ShapeUtil::Compatible(reducer.xla_output_shape, expected_shape), in Compile()
102 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
113 xla::Reduce(context->builder(), inputs, inits, *reducer.computation, in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc586 Reducer reducer; member
600 (param.reducer == kAdd) ? "_add" : "_max"); in R4ReduceWindowTestDataToString()
601 CHECK(param.reducer == kAdd || param.reducer == kMax); in R4ReduceWindowTestDataToString()
649 CHECK(param.reducer == kAdd || param.reducer == kMax); in DoIt()
650 auto reducer = param.reducer; in DoIt() local
651 auto computation = reducer == kAdd in DoIt()
664 CHECK(reducer == kAdd || reducer == kMax); in DoIt()
665 auto reduce_func = reducer == kAdd in DoIt()
956 Reducer reducer; member
1012 param.reducer == kAdd ? "add" : "max"); in R3ReduceWindowTestDataToString()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc93 XlaComputation reducer = in CreateScalarIdentityWithZeroComputation() local
97 return reducer; in CreateScalarIdentityWithZeroComputation()
161 XlaComputation reducer = in ArgMinMax() local
165 {value_init_value, index_init_value}, reducer, in ArgMinMax()
181 XlaComputation reducer; in ArgMinMaxTwoPass() local
184 reducer = CreateScalarMinComputation(input_shape.element_type(), builder); in ArgMinMaxTwoPass()
187 reducer = CreateScalarMaxComputation(input_shape.element_type(), builder); in ArgMinMaxTwoPass()
192 XlaOp reduced_input = Reduce(input, init_value, reducer, in ArgMinMaxTwoPass()
/external/rust/crates/rayon/src/iter/find_first_last/
Dtest.rs21 let reducer = consumer.to_reducer(); in same_range_first_consumers_return_correct_answer() localVariable
35 reducer.reduce(left_folder.complete(), right_folder.complete()), in same_range_first_consumers_return_correct_answer()
56 let reducer = consumer.to_reducer(); in same_range_last_consumers_return_correct_answer() localVariable
71 reducer.reduce(left_folder.complete(), right_folder.complete()), in same_range_last_consumers_return_correct_answer()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExperimentalGroupByReducerDataset.pbtxt35 `init_func_other_arguments` to the initial reducer state.
48 A function mapping the current reducer state and an element of `input_dataset`,
49 concatenated with `reduce_func_other_arguments` to a new reducer state.
62 A function mapping the final reducer state to an output element.
Dapi_def_GroupByReducerDataset.pbtxt35 `init_func_other_arguments` to the initial reducer state.
48 A function mapping the current reducer state and an element of `input_dataset`,
49 concatenated with `reduce_func_other_arguments` to a new reducer state.
62 A function mapping the final reducer state to an output element.
/external/rust/crates/rayon/src/iter/
Dflatten.rs65 let (left, right, reducer) = self.base.split_at(index); in split_at()
69 reducer, in split_at()
119 let reducer = self.base.to_reducer(); in consume() localVariable
120 Some(reducer.reduce(previous, result)) in consume()

123456