Home
last modified time | relevance | path

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

1234

/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/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/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/deqp-deps/SPIRV-Tools/test/reduce/
Dreducer_test.cpp213 Reducer reducer(env); in TEST() local
215 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
216 reducer.SetInterestingnessFunction( in TEST()
220 reducer.AddReductionPass(MakeUnique<OperandToConstReductionPass>(env)); in TEST()
221 reducer.AddReductionPass( in TEST()
232 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
281 Reducer reducer(env); in TEST() local
285 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
286 reducer.SetInterestingnessFunction( in TEST()
290 reducer.AddReductionPass( in TEST()
[all …]
Dvalidation_during_reduction_test.cpp149 Reducer reducer(env); in TEST() local
150 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
153 reducer.SetInterestingnessFunction( in TEST()
156 reducer.AddReductionPass(MakeUnique<BlindlyRemoveGlobalValuesPass>(env)); in TEST()
166 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
353 Reducer reducer(env); in TEST() local
354 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
357 reducer.SetInterestingnessFunction( in TEST()
360 reducer.AddReductionPass(MakeUnique<BlindlyRemoveGlobalValuesPass>(env)); in TEST()
370 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
/external/swiftshader/third_party/SPIRV-Tools/test/reduce/
Dreducer_test.cpp213 Reducer reducer(env); in TEST() local
215 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
216 reducer.SetInterestingnessFunction( in TEST()
220 reducer.AddReductionPass(MakeUnique<OperandToConstReductionPass>(env)); in TEST()
221 reducer.AddReductionPass( in TEST()
232 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
281 Reducer reducer(env); in TEST() local
285 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
286 reducer.SetInterestingnessFunction( in TEST()
290 reducer.AddReductionPass( in TEST()
[all …]
Dvalidation_during_reduction_test.cpp149 Reducer reducer(env); in TEST() local
150 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
153 reducer.SetInterestingnessFunction( in TEST()
156 reducer.AddReductionPass(MakeUnique<BlindlyRemoveGlobalValuesPass>(env)); in TEST()
166 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
353 Reducer reducer(env); in TEST() local
354 reducer.SetMessageConsumer(NopDiagnostic); in TEST()
357 reducer.SetInterestingnessFunction( in TEST()
360 reducer.AddReductionPass(MakeUnique<BlindlyRemoveGlobalValuesPass>(env)); in TEST()
370 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in TEST()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py40 reducer = grouping.Reducer(
46 grouping.group_by_reducer(lambda x: x % 2, reducer))
58 reducer = grouping.Reducer(
65 lambda x: math_ops.cast(x, dtypes.int64) % 2, reducer))
73 reducer = grouping.Reducer(
81 grouping.group_by_reducer(lambda x, y: y % 2, reducer))
94 reducer = grouping.Reducer(
100 grouping.group_by_reducer(lambda x: x.values[0] % 2, reducer))
115 reducer = grouping.Reducer(
122 grouping.group_by_reducer(lambda x: x, reducer))
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops.h46 const ReductionAxes& reduction_axes, const Reducer& reducer) { in operator()
47 out.device(d) = in.reduce(reduction_axes, reducer); in operator()
57 const functor::MeanReducer<Scalar>& reducer) {
73 const functor::EuclideanNormReducer<Scalar>& reducer) {
87 const functor::EuclideanNormReducer<bfloat16>& reducer) {
101 static auto identity(const Reducer& reducer)
102 -> decltype(reducer.initialize()) {
103 return reducer.initialize();
125 void FillIdentityEigenImpl(const Device& d, OUT_T out, const Reducer& reducer) {
126 out.device(d) = out.constant(Identity<Reducer>::identity(reducer));
[all …]
Dreduction_ops_common.h185 Reducer reducer;
194 Functor::FillIdentity(d, tmp_out.flat<T>(), reducer);
198 constants.kZero, reducer);
202 constants.kZero, reducer);
206 constants.kOne, reducer);
211 constants.kZeroTwo, reducer);
215 constants.kOne, reducer);
232 constants.kOne, reducer);
257 const Reducer& reducer) {
260 reducer_impl(d, out, in, reduction_axes, reducer);
[all …]
Dreduction_gpu_kernels.cu.h889 const Reducer& reducer);
897 const Eigen::internal::SumReducer<T>& reducer) {
907 const Eigen::internal::SumReducer<T>& reducer) {
908 FillIdentityEigenImpl(d, To32Bit(out), reducer);
918 const functor::EuclideanNormReducer<T>& reducer) {
932 const functor::EuclideanNormReducer<T>& reducer) {
933 FillIdentityEigenImpl(d, To32Bit(out), reducer);
942 const functor::MeanReducer<T>& reducer) {
968 const functor::MeanReducer<T>& reducer) {
969 FillIdentityEigenImpl(d, To32Bit(out), reducer);
[all …]
/external/swiftshader/third_party/SPIRV-Tools/tools/reduce/
Dreduce.cpp191 Reducer reducer(target_env); in main() local
193 reducer.SetInterestingnessFunction( in main()
209 reducer.AddReductionPass( in main()
211 reducer.AddReductionPass( in main()
213 reducer.AddReductionPass( in main()
215 reducer.AddReductionPass( in main()
217 reducer.AddReductionPass( in main()
220 reducer.AddReductionPass( in main()
223 reducer.SetMessageConsumer(spvtools::utils::CLIMessageConsumer); in main()
232 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in main()
/external/deqp-deps/SPIRV-Tools/tools/reduce/
Dreduce.cpp190 Reducer reducer(target_env); in main() local
192 reducer.SetInterestingnessFunction( in main()
208 reducer.AddReductionPass( in main()
210 reducer.AddReductionPass( in main()
212 reducer.AddReductionPass( in main()
214 reducer.AddReductionPass( in main()
217 reducer.AddReductionPass( in main()
220 reducer.SetMessageConsumer(spvtools::utils::CLIMessageConsumer); in main()
229 reducer.Run(std::move(binary_in), &binary_out, reducer_options); in main()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc571 Reducer reducer; member
585 (param.reducer == kAdd) ? "_add" : "_max"); in R4ReduceWindowTestDataToString()
586 CHECK(param.reducer == kAdd || param.reducer == kMax); in R4ReduceWindowTestDataToString()
633 CHECK(param.reducer == kAdd || param.reducer == kMax); in DoIt()
634 auto reducer = param.reducer; in DoIt() local
635 auto computation = reducer == kAdd in DoIt()
648 CHECK(reducer == kAdd || reducer == kMax); in DoIt()
649 auto reduce_func = reducer == kAdd in DoIt()
940 Reducer reducer; member
996 param.reducer == kAdd ? "add" : "max"); in R3ReduceWindowTestDataToString()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_reduce_op.cc70 XlaCompiler::CompilationResult reducer; in Compile() local
73 {reducer_arg, reducer_arg}, &reducer)); in Compile()
80 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
84 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
88 *reducer.computation, dimensions_to_reduce_); in Compile()
Dreduce_window_op.cc87 XlaCompiler::CompilationResult reducer; in Compile() local
90 {reducer_arg, reducer_arg}, &reducer)); in Compile()
97 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
101 xla::ShapeUtil::HumanString(reducer.xla_output_shape))); in Compile()
120 context->Input(0), context->Input(1), *reducer.computation, in Compile()
Dscan_ops.cc96 const xla::XlaComputation* reducer; in Compile() local
99 reducer = ctx->GetOrCreateAdd(dtype); in Compile()
102 reducer = ctx->GetOrCreateMul(dtype); in Compile()
105 XlaHelpers::ConvertElementType(ctx->Input(0), dtype), init, *reducer, in Compile()
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dget_single_element.py68 def reduce_dataset(dataset, reducer): argument
85 return dataset.reduce(reducer.init_func(np.int64(0)), reducer.reduce_func)
/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.
/external/skia/src/pathops/
DSkReduceOrder.cpp249 SkReduceOrder reducer; in Quad() local
250 int order = reducer.reduce(quad); in Quad()
253 *reducePts++ = reducer.fLine[index].asSkPoint(); in Quad()
275 SkReduceOrder reducer; in Cubic() local
276 int order = reducer.reduce(cubic, kAllow_Quadratics); in Cubic()
279 *reducePts++ = reducer.fQuad[index].asSkPoint(); in Cubic()
/external/skqp/src/pathops/
DSkReduceOrder.cpp249 SkReduceOrder reducer; in Quad() local
250 int order = reducer.reduce(quad); in Quad()
253 *reducePts++ = reducer.fLine[index].asSkPoint(); in Quad()
275 SkReduceOrder reducer; in Cubic() local
276 int order = reducer.reduce(cubic, kAllow_Quadratics); in Cubic()
279 *reducePts++ = reducer.fQuad[index].asSkPoint(); in Cubic()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc133 XlaComputation reducer; in ArgMinMax() local
136 reducer = CreateScalarMinComputation(input_shape.element_type(), builder); in ArgMinMax()
139 reducer = CreateScalarMaxComputation(input_shape.element_type(), builder); in ArgMinMax()
142 XlaOp input_max = Reduce(input, init_value, reducer, in ArgMinMax()

1234