| /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 | arithmetic.cc | 32 using XlaOpGenerator = XlaOp (*)(XlaBuilder*, const XlaOp&, const XlaOp&);
35 XlaBuilder* builder, in CreateScalarComputation()
37 std::unique_ptr<XlaBuilder> b; in CreateScalarComputation()
55 XlaBuilder* builder) { in CreateScalarAddComputation()
58 [](XlaBuilder* b, const XlaOp& lhs, const XlaOp& rhs) { in CreateScalarAddComputation()
64 XlaBuilder* builder) { in CreateScalarMultiplyComputation()
67 [](XlaBuilder* b, const XlaOp& lhs, const XlaOp& rhs) { in CreateScalarMultiplyComputation()
73 XlaBuilder* builder) { in CreateScalarGeComputation()
75 [](XlaBuilder* b, const XlaOp& lhs, in CreateScalarGeComputation()
80 XlaBuilder* builder) { in CreateScalarMaxComputation()
[all …]
|
| D | constants.h | 34 XlaOp ConstantR0WithType(XlaBuilder* builder, PrimitiveType type, T value) { in ConstantR0WithType()
86 XlaBuilder* builder = prototype.builder(); in ScalarLike()
100 XlaBuilder* builder = prototype.builder(); in FullLike()
115 XlaOp Zero(XlaBuilder* builder, PrimitiveType type);
118 XlaOp Zeros(XlaBuilder* builder, const Shape& shape);
124 XlaOp One(XlaBuilder* builder, PrimitiveType type);
128 XlaOp Epsilon(XlaBuilder* builder, PrimitiveType type);
132 XlaOp MinValue(XlaBuilder* builder, PrimitiveType type);
136 XlaOp MinFiniteValue(XlaBuilder* builder, PrimitiveType type);
139 XlaOp MinPositiveNormalValue(XlaBuilder* builder, PrimitiveType type);
[all …]
|
| D | math_test.cc | 43 XlaBuilder b(TestName()); in TestLogEdgeCases()
54 XlaBuilder b(TestName()); in TestLog1pEdgeCases()
64 XlaBuilder b(TestName()); in TestIsInfOrNan()
95 XlaBuilder b(TestName()); in TestIsNegZero()
130 XlaBuilder b(TestName()); in TestSqrtPowInequivalence()
142 XlaBuilder b(TestName()); in TestErfEdgeCases()
205 XlaBuilder b(TestName()); in XLA_TEST_F()
215 XlaBuilder builder(TestName()); in XLA_TEST_F()
228 XlaBuilder builder(TestName()); in XLA_TEST_F()
239 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | arithmetic.h | 29 XlaBuilder* builder);
33 XlaBuilder* builder);
37 XlaBuilder* builder);
41 XlaBuilder* builder);
45 XlaBuilder* builder);
49 XlaBuilder* builder);
53 XlaBuilder* builder);
|
| 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 | 53 XlaBuilder builder(TestName()); in XLA_TEST_F()
61 XlaBuilder builder(TestName()); in XLA_TEST_F()
70 XlaBuilder builder(TestName()); in XLA_TEST_F()
86 XlaBuilder builder(TestName()); in XLA_TEST_F()
94 XlaBuilder builder(TestName()); in XLA_TEST_F()
105 XlaBuilder builder(TestName()); in XLA_TEST_F()
133 XlaBuilder builder(TestName()); in XLA_TEST_F()
144 XlaBuilder builder(TestName()); in XLA_TEST_F()
164 XlaBuilder builder(TestName()); in XLA_TEST_F()
176 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| D | client_library_test_base.h | 96 XlaBuilder* builder, absl::Span<GlobalData* const> arguments);
99 XlaBuilder* builder, absl::Span<GlobalData* const> arguments,
117 string ExecuteToString(XlaBuilder* builder,
127 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
130 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
135 void ComputeAndCompareR1(XlaBuilder* builder,
139 void ComputeAndCompareR1(XlaBuilder* builder,
146 void ComputeAndCompareR1(XlaBuilder* builder,
151 void ComputeAndCompareR2(XlaBuilder* builder,
155 void ComputeAndCompareR2(XlaBuilder* builder,
[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 | 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 | 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 | reshape_test.cc | 55 XlaBuilder builder(TestName()); in XLA_TEST_P()
70 XlaBuilder builder(TestName()); in XLA_TEST_P()
83 XlaBuilder builder(TestName()); in XLA_TEST_P()
97 XlaBuilder builder(TestName()); in XLA_TEST_P()
114 XlaBuilder builder(TestName()); in XLA_TEST_P()
129 XlaBuilder builder(TestName()); in XLA_TEST_P()
142 XlaBuilder builder(TestName()); in XLA_TEST_P()
156 XlaBuilder builder(TestName()); in XLA_TEST_P()
170 XlaBuilder builder(TestName()); in XLA_TEST_P()
183 XlaBuilder builder(TestName()); in XLA_TEST_P()
[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 | 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 | 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 | concat_test.cc | 41 XlaBuilder builder(TestName()); in XLA_TEST_F()
51 XlaBuilder builder(TestName()); in XLA_TEST_F()
60 XlaBuilder builder(TestName()); in XLA_TEST_F()
71 XlaBuilder builder(TestName()); in XLA_TEST_F()
82 XlaBuilder builder(TestName()); in XLA_TEST_F()
92 XlaBuilder builder(TestName()); in XLA_TEST_F()
102 XlaBuilder builder(TestName()); in XLA_TEST_F()
112 XlaBuilder builder(TestName()); in XLA_TEST_F()
132 XlaBuilder builder(TestName()); in XLA_TEST_F()
142 XlaBuilder builder(TestName()); in XLA_TEST_F()
[all …]
|
| 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 | 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 | map_test.cc | 54 XlaBuilder mapped_builder(TestName()); in CreateAdderToOne()
64 XlaBuilder b(TestName()); in CreateMax()
77 XlaBuilder mapped_builder("scalar_one"); in CreateScalarOne()
91 XlaBuilder mapped_builder(TestName()); in CreateMulByTwo()
109 XlaBuilder mapped_builder(TestName()); in CreateAdderToOneTimesItself()
127 XlaBuilder builder(TestName()); in CreateMapPlusN()
140 XlaBuilder b("Gt"); in CreateGt()
157 XlaBuilder mapped_builder("TernaryAdder"); in CreateTernaryAdder()
171 XlaBuilder builder(TestName()); in TEST_F()
185 XlaBuilder builder(TestName()); in XLA_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()
89 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
93 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
108 xla::XlaOp InitialValue(xla::XlaBuilder* builder) override { in InitialValue()
111 void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, in BuildReducer()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 42 StatusOr<std::unique_ptr<HloModule>> BuildHloModule(XlaBuilder* b) { in BuildHloModule()
53 StatusOr<std::unique_ptr<HloModule>> BuildHloModule(XlaBuilder* b, in BuildHloModule()
71 XlaBuilder b(TestName()); in TEST_F()
82 XlaBuilder b(TestName()); in TEST_F()
96 XlaBuilder b(TestName()); in TEST_F()
127 XlaBuilder b(TestName()); in TEST_F()
139 XlaBuilder b(TestName()); in TEST_F()
155 XlaBuilder b(TestName()); in TEST_F()
171 XlaBuilder b(TestName()); in TEST_F()
181 XlaBuilder b(TestName()); in TEST_F()
[all …]
|