Searched defs:sum_reducer (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | vector_ops_reduce_test.cc | 57 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
68 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
104 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
120 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
134 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
145 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
157 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
168 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
177 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in XLA_TEST_F() local
186 auto sum_reducer = CreateScalarAddComputation(F32, &builder_); in TEST_F() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | reduce_window_test.py | 44 def sum_reducer(x, y): function
|
| H A D | xla_ops_test.py | 343 def sum_reducer(x, y): function
|
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
| H A D | microbatch.py | 49 sum_reducer = _LossReducer(torch.tensor(0.0), lambda a, b: a + b) variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scan_ops.h | 119 auto sum_reducer = Eigen::internal::SumReducer<T>(); in finalizeBoth() local
|