| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | constants_test.cc | 32 XlaBuilder builder(TestName()); in XLA_TEST_F()
38 XlaBuilder builder(TestName()); in XLA_TEST_F()
46 XlaBuilder builder(TestName()); in XLA_TEST_F()
54 XlaBuilder builder(TestName()); in XLA_TEST_F()
60 XlaBuilder builder(TestName()); in XLA_TEST_F()
66 XlaBuilder builder(TestName()); in XLA_TEST_F()
72 XlaBuilder builder(TestName()); in XLA_TEST_F()
78 XlaBuilder builder(TestName()); in XLA_TEST_F()
84 XlaBuilder builder(TestName()); in XLA_TEST_F()
90 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | math_test.cc | 46 XlaBuilder b(TestName()); in TestLogEdgeCases()
57 XlaBuilder b(TestName()); in TestLog1pEdgeCases()
67 XlaBuilder b(TestName()); in TestIsInfOrNan()
98 XlaBuilder b(TestName()); in TestIsNegZero()
137 XlaBuilder b(TestName()); in TestSqrtPowInequivalence()
149 XlaBuilder b(TestName()); in TestErfEdgeCases()
217 XlaBuilder b(TestName()); in XLA_TEST_F()
227 XlaBuilder builder(TestName()); in XLA_TEST_F()
240 XlaBuilder builder(TestName()); in XLA_TEST_F()
254 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | arithmetic.h | 31 XlaBuilder* builder,
36 XlaBuilder* builder);
40 XlaBuilder* builder);
44 XlaBuilder* builder);
48 XlaBuilder* builder);
52 XlaBuilder* builder);
56 XlaBuilder* builder);
60 XlaBuilder* builder);
72 XlaBuilder* builder);
|
| D | sorting_test.cc | 32 XlaBuilder builder(TestName()); in XLA_TEST_F()
40 XlaBuilder builder(TestName()); in XLA_TEST_F()
48 XlaBuilder builder(TestName()); in XLA_TEST_F()
57 XlaBuilder builder(TestName()); in XLA_TEST_F()
72 XlaBuilder builder(TestName()); in XLA_TEST_F()
80 XlaBuilder builder(TestName()); in XLA_TEST_F()
88 XlaBuilder builder(TestName()); in XLA_TEST_F()
97 XlaBuilder builder(TestName()); in XLA_TEST_F()
105 XlaBuilder builder(TestName()); in XLA_TEST_F()
114 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | constants.h | 34 XlaOp ConstantR0WithType(XlaBuilder* builder, PrimitiveType type, T value) { in ConstantR0WithType()
92 XlaBuilder* builder = prototype.builder(); in ScalarLike()
106 XlaBuilder* builder = prototype.builder(); in FullLike()
121 XlaOp Zero(XlaBuilder* builder, PrimitiveType type);
124 XlaOp Zeros(XlaBuilder* builder, const Shape& shape);
130 XlaOp One(XlaBuilder* builder, PrimitiveType type);
134 XlaOp Epsilon(XlaBuilder* builder, PrimitiveType type);
138 XlaOp MinValue(XlaBuilder* builder, PrimitiveType type);
142 XlaOp MinFiniteValue(XlaBuilder* builder, PrimitiveType type);
145 XlaOp MinPositiveNormalValue(XlaBuilder* builder, PrimitiveType type);
[all …]
|
| D | constants.cc | 23 XlaOp Zero(XlaBuilder* builder, PrimitiveType type) { in Zero()
27 XlaOp Zeros(XlaBuilder* builder, const Shape& shape) { in Zeros()
33 XlaBuilder* builder = prototype.builder(); in ZerosLike()
40 XlaOp One(XlaBuilder* builder, PrimitiveType type) { in One()
44 XlaOp Epsilon(XlaBuilder* builder, PrimitiveType type) { in Epsilon()
65 XlaOp MinValue(XlaBuilder* builder, PrimitiveType type) { in MinValue()
69 XlaOp MinFiniteValue(XlaBuilder* builder, PrimitiveType type) { in MinFiniteValue()
86 XlaOp MinPositiveNormalValue(XlaBuilder* builder, PrimitiveType type) { in MinPositiveNormalValue()
105 XlaOp MaxValue(XlaBuilder* builder, PrimitiveType type) { in MaxValue()
109 XlaOp MaxFiniteValue(XlaBuilder* builder, PrimitiveType type) { in MaxFiniteValue()
[all …]
|
| D | pooling_test.cc | 38 XlaBuilder* b = input.builder(); in MakeGeneralPadding()
66 XlaBuilder builder(TestName()); in XLA_TEST_F()
79 XlaBuilder builder(TestName()); in XLA_TEST_F()
92 XlaBuilder builder(TestName()); in XLA_TEST_F()
106 XlaBuilder builder(TestName()); in XLA_TEST_F()
122 XlaBuilder builder(TestName()); in XLA_TEST_F()
138 XlaBuilder builder(TestName()); in XLA_TEST_F()
156 XlaBuilder builder(TestName()); in XLA_TEST_F()
171 XlaBuilder builder(TestName()); in XLA_TEST_F()
186 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 54 XlaBuilder builder(TestName()); in XLA_TEST_F()
62 XlaBuilder builder(TestName()); in XLA_TEST_F()
71 XlaBuilder builder(TestName()); in XLA_TEST_F()
79 XlaBuilder builder(TestName()); in XLA_TEST_F()
95 XlaBuilder builder(TestName()); in XLA_TEST_F()
103 XlaBuilder builder(TestName()); in XLA_TEST_F()
114 XlaBuilder builder(TestName()); in XLA_TEST_F()
142 XlaBuilder builder(TestName()); in XLA_TEST_F()
150 XlaBuilder builder(TestName()); in XLA_TEST_F()
163 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | client_library_test_base.h | 98 XlaBuilder* builder, absl::Span<GlobalData* const> arguments);
101 XlaBuilder* builder, absl::Span<GlobalData* const> arguments,
119 string ExecuteToString(XlaBuilder* builder,
129 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
132 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
137 void ComputeAndCompareR1(XlaBuilder* builder,
141 void ComputeAndCompareR1(XlaBuilder* builder,
148 void ComputeAndCompareR1(XlaBuilder* builder,
153 void ComputeAndCompareR2(XlaBuilder* builder,
157 void ComputeAndCompareR2(XlaBuilder* builder,
[all …]
|
| D | scalar_computations_test.cc | 51 XlaBuilder builder(TestName()); in TestCompare()
62 XlaBuilder builder(TestName()); in TestMinMax()
71 XlaBuilder builder(TestName()); in XLA_TEST_F()
78 XlaBuilder builder(TestName()); in XLA_TEST_F()
85 XlaBuilder builder(TestName()); in XLA_TEST_F()
92 XlaBuilder builder(TestName()); in XLA_TEST_F()
99 XlaBuilder builder(TestName()); in XLA_TEST_F()
106 XlaBuilder builder(TestName()); in XLA_TEST_F()
113 XlaBuilder builder(TestName()); in XLA_TEST_F()
120 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | constants_test.cc | 45 XlaBuilder builder(TestName()); in TEST_F()
54 XlaBuilder builder(TestName()); in TEST_F()
63 XlaBuilder builder(TestName()); in TEST_F()
72 XlaBuilder builder(TestName()); in TEST_F()
81 XlaBuilder builder(TestName()); in TEST_F()
91 XlaBuilder builder(TestName()); in TEST_F()
98 XlaBuilder builder(TestName()); in TEST_F()
108 XlaBuilder builder(TestName()); in TEST_F()
115 XlaBuilder builder(TestName()); in TEST_F()
123 XlaBuilder builder(TestName()); in TEST_F()
[all …]
|
| D | while_test.cc | 57 XlaBuilder builder("condition"); in XLA_TEST_F()
66 XlaBuilder builder("body"); in XLA_TEST_F()
74 XlaBuilder builder(TestName()); in XLA_TEST_F()
93 XlaBuilder builder("condition"); in XLA_TEST_F()
102 XlaBuilder builder("body"); in XLA_TEST_F()
110 XlaBuilder builder(TestName()); in XLA_TEST_F()
124 XlaBuilder builder("condition"); in XLA_TEST_F()
133 XlaBuilder builder("body"); in XLA_TEST_F()
141 XlaBuilder builder(TestName()); in XLA_TEST_F()
156 XlaBuilder builder("condition"); in XLA_TEST_F()
[all …]
|
| D | convert_test.cc | 49 XlaBuilder builder(TestName()); in TEST_F()
58 XlaBuilder builder(TestName()); in TEST_F()
67 XlaBuilder builder(TestName()); in TEST_F()
76 XlaBuilder builder(TestName()); in TEST_F()
85 XlaBuilder builder(TestName()); in TEST_F()
94 XlaBuilder builder(TestName()); in TEST_F()
103 XlaBuilder builder(TestName()); in TEST_F()
112 XlaBuilder builder(TestName()); in TEST_F()
121 XlaBuilder builder(TestName()); in TEST_F()
130 XlaBuilder builder(TestName()); in TEST_F()
[all …]
|
| D | vector_ops_simple_test.cc | 52 XlaBuilder builder(TestName()); in XLA_TEST_F()
66 XlaBuilder builder(TestName()); in XLA_TEST_F()
87 XlaBuilder builder(TestName()); in XLA_TEST_F()
109 XlaBuilder builder(TestName()); in XLA_TEST_F()
120 XlaBuilder builder(TestName()); in XLA_TEST_F()
129 XlaBuilder builder(TestName()); in XLA_TEST_F()
139 XlaBuilder builder(TestName()); in XLA_TEST_F()
151 XlaBuilder builder(TestName()); in XLA_TEST_F()
166 XlaBuilder builder(TestName()); in XLA_TEST_F()
181 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | convolution_variants_test.cc | 55 XlaBuilder builder(TestName()); in XLA_TEST_F()
70 XlaBuilder builder(TestName()); in XLA_TEST_F()
85 XlaBuilder builder(TestName()); in XLA_TEST_F()
102 XlaBuilder builder(TestName()); in XLA_TEST_F()
117 XlaBuilder builder(TestName()); in XLA_TEST_F()
132 XlaBuilder builder(TestName()); in XLA_TEST_F()
147 XlaBuilder builder(TestName()); in XLA_TEST_F()
162 XlaBuilder builder(TestName()); in XLA_TEST_F()
177 XlaBuilder builder(TestName()); in XLA_TEST_F()
192 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | pred_test.cc | 36 XlaBuilder builder(TestName()); in TestCompare()
45 XlaBuilder builder(TestName()); in TEST_F()
51 XlaBuilder builder(TestName()); in TEST_F()
81 XlaBuilder builder(TestName()); in TEST_F()
87 XlaBuilder builder(TestName()); in TEST_F()
97 XlaBuilder builder(TestName()); in TEST_F()
104 XlaBuilder builder(TestName()); in TEST_F()
111 XlaBuilder builder(TestName()); in TEST_F()
118 XlaBuilder builder(TestName()); in TEST_F()
129 XlaBuilder builder(TestName()); in TEST_F()
|
| D | transpose_test.cc | 40 XlaBuilder builder("Transpose"); in XLA_TEST_F()
48 XlaBuilder builder("Transpose"); in XLA_TEST_F()
56 XlaBuilder builder("Transpose"); in XLA_TEST_F()
64 XlaBuilder builder("Transpose"); in TEST_F()
77 XlaBuilder builder("Transpose"); in XLA_TEST_F()
86 XlaBuilder builder("Transpose"); in TEST_F()
97 XlaBuilder builder("Transpose"); in TEST_F()
108 XlaBuilder builder("Transpose"); in TEST_F()
123 XlaBuilder builder("Transpose"); in TEST_F()
137 XlaBuilder builder("transpose_1x1"); in TEST_F()
[all …]
|
| D | conditional_test.cc | 29 XlaBuilder builder("Constant"); in CreateR0ConstantComputation()
38 XlaBuilder builder("Identity"); in CreateR0IdentityComputation()
46 XlaBuilder builder("Ceil"); in CreateCeilComputation()
63 XlaBuilder builder("Floor"); in CreateFloorComputation()
81 XlaBuilder builder(computation_name); in CreateTupleCeilComputation()
103 XlaBuilder builder(computation_name); in CreateTupleFloorComputation()
125 XlaBuilder builder(computation_name); in CreateTupleAddComputation()
145 XlaBuilder builder(computation_name); in CreateTupleSubComputation()
180 XlaBuilder builder(TestName()); in XLA_TEST_F()
196 XlaBuilder builder(TestName()); in XLA_TEST_P()
[all …]
|
| D | concat_test.cc | 44 XlaBuilder builder(TestName()); in XLA_TEST_F()
54 XlaBuilder builder(TestName()); in XLA_TEST_F()
63 XlaBuilder builder(TestName()); in XLA_TEST_F()
74 XlaBuilder builder(TestName()); in XLA_TEST_F()
85 XlaBuilder builder(TestName()); in XLA_TEST_F()
95 XlaBuilder builder(TestName()); in XLA_TEST_F()
105 XlaBuilder builder(TestName()); in XLA_TEST_F()
115 XlaBuilder builder(TestName()); in XLA_TEST_F()
135 XlaBuilder builder(TestName()); in XLA_TEST_F()
145 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | broadcast_simple_test.cc | 38 XlaBuilder* builder) { in BuildBinOp()
109 XlaBuilder b(TestName()); in XLA_TEST_F()
115 XlaBuilder b(TestName()); in XLA_TEST_F()
122 XlaBuilder b(TestName()); in XLA_TEST_F()
135 XlaBuilder b(TestName()); in XLA_TEST_F()
142 XlaBuilder b(TestName()); in XLA_TEST_F()
149 XlaBuilder b(TestName());
163 XlaBuilder b(TestName());
176 XlaBuilder b(TestName());
189 XlaBuilder b(TestName());
[all …]
|
| D | reshape_test.cc | 55 XlaBuilder builder(TestName()); in XLA_TEST_P()
71 XlaBuilder builder(TestName()); in XLA_TEST_P()
85 XlaBuilder builder(TestName()); in XLA_TEST_P()
100 XlaBuilder builder(TestName()); in XLA_TEST_P()
118 XlaBuilder builder(TestName()); in XLA_TEST_P()
134 XlaBuilder builder(TestName()); in XLA_TEST_P()
148 XlaBuilder builder(TestName()); in XLA_TEST_P()
163 XlaBuilder builder(TestName()); in XLA_TEST_P()
178 XlaBuilder builder(TestName()); in XLA_TEST_P()
192 XlaBuilder builder(TestName()); in XLA_TEST_P()
[all …]
|
| D | select_test.cc | 37 XlaBuilder builder(TestName()); in TEST_F()
47 XlaBuilder builder(TestName()); in TEST_F()
57 XlaBuilder builder(TestName()); in TEST_F()
67 XlaBuilder builder(TestName()); in XLA_TEST_F()
77 XlaBuilder builder(TestName()); in TEST_F()
92 XlaBuilder builder(TestName()); in XLA_TEST_F()
106 XlaBuilder builder(TestName()); in TEST_F()
122 XlaBuilder builder(TestName()); in TEST_F()
139 XlaBuilder builder(TestName()); in TEST_F()
159 XlaBuilder builder(TestName()); in TEST_F()
[all …]
|
| D | call_test.cc | 37 XlaBuilder builder("Identity"); in CreateR0F32IdentityComputation()
45 XlaBuilder builder("Addition"); in CreateR1S0F32AdditionComputation()
55 XlaBuilder builder("Addition"); in CreateR1S2F32AdditionComputation()
65 XlaBuilder builder("Tuple"); in CreateR0F32TupleComputation()
78 XlaBuilder builder(TestName()); in XLA_TEST_F()
87 XlaBuilder builder(TestName()); in XLA_TEST_F()
97 XlaBuilder builder(TestName()); in XLA_TEST_F()
109 XlaBuilder builder("inner"); in XLA_TEST_F()
116 XlaBuilder builder2("outer"); in XLA_TEST_F()
125 XlaBuilder builder3("outermost"); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops.cc | 35 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
38 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
53 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
57 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
71 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
75 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
103 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
107 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
122 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
125 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 44 StatusOr<std::unique_ptr<HloModule>> BuildHloModule(XlaBuilder* b) { in BuildHloModule()
55 StatusOr<std::unique_ptr<HloModule>> BuildHloModule(XlaBuilder* b, in BuildHloModule()
73 XlaBuilder b(TestName()); in TEST_F()
84 XlaBuilder b(TestName()); in TEST_F()
98 XlaBuilder b(TestName()); in TEST_F()
129 XlaBuilder b(TestName()); in TEST_F()
141 XlaBuilder b(TestName()); in TEST_F()
157 XlaBuilder b(TestName()); in TEST_F()
173 XlaBuilder b(TestName()); in TEST_F()
183 XlaBuilder b(TestName()); in TEST_F()
[all …]
|