| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| vector_ops_reduce_test.cc | 57 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
69 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
108 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
125 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
140 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
152 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
165 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
177 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
187 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
197 auto sum_reducer = CreateScalarAddComputation(F32, &builder_)
[all...] |
| reduce_test.cc | 85 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
198 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
226 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
438 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
469 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
501 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
521 Computation add_f32 = CreateScalarAddComputation(F32, &builder);
568 auto add = CreateScalarAddComputation(F32, &builder);
650 auto add = CreateScalarAddComputation(F32, &builder);
661 auto add = CreateScalarAddComputation(F32, &builder)
[all...] |
| reduce_window_test.cc | 77 CreateScalarAddComputation(FloatType(), &builder_),
110 CreateScalarAddComputation(FloatType(), &builder_),
126 CreateScalarAddComputation(FloatType(), &builder_),
595 ? CreateScalarAddComputation(FloatType(), &b)
[all...] |
| select_and_scatter_test.cc | 56 add_s32_ = CreateScalarAddComputation(S32, &builder_);
58 add_f32_ = CreateScalarAddComputation(F32, &builder_);
|
| batch_normalization_test.cc | 129 Computation add = CreateScalarAddComputation(F32, &builder);
144 Computation add = CreateScalarAddComputation(F32, &builder);
171 Computation add = CreateScalarAddComputation(F32, &builder);
[all...] |
| map_test.cc | 397 CreateScalarAddComputation(F32, &builder), {0});
421 CreateScalarAddComputation(S32, &builder), {0, 1});
447 CreateScalarAddComputation(S32, &builder), {0, 1, 2});
|
| vector_ops_simple_test.cc | 192 auto add = CreateScalarAddComputation(F32, &builder);
|
| pad_test.cc | 452 Computation add = CreateScalarAddComputation(FloatType(), &b);
|
| while_test.cc | 148 CreateScalarAddComputation(S32, &builder), {0});
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| arithmetic.h | 28 Computation CreateScalarAddComputation(PrimitiveType type,
|
| arithmetic.cc | 52 Computation CreateScalarAddComputation(PrimitiveType type,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| pooling_ops.cc | 346 auto scatter = CreateScalarAddComputation(element_type, ctx->builder());
|
