| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.h | 29 ComputationBuilder* builder);
33 ComputationBuilder* builder);
37 ComputationBuilder* builder);
41 ComputationBuilder* builder);
45 ComputationBuilder* builder);
48 Computation CreateScalarAndComputation(ComputationBuilder* builder);
51 Computation CreateScalarOrComputation(ComputationBuilder* builder);
57 ComputationBuilder* builder);
|
| H A D | arithmetic.cc | 30 ComputationDataHandle (*)(ComputationBuilder*, const ComputationDataHandle&,
34 ComputationBuilder* builder,
36 std::unique_ptr<ComputationBuilder> b;
53 ComputationBuilder* builder) {
56 [](ComputationBuilder* b, const ComputationDataHandle& lhs,
61 ComputationBuilder* builder) {
64 [](ComputationBuilder* b, const ComputationDataHandle& lhs,
69 ComputationBuilder* builder) {
72 [](ComputationBuilder* b, const ComputationDataHandle& lhs,
77 ComputationBuilder* builde
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | pred_test.cc | 33 ComputationDataHandle (ComputationBuilder::*op)(
37 ComputationBuilder builder(client_, TestName());
46 ComputationBuilder builder(client_, TestName());
52 ComputationBuilder builder(client_, TestName());
58 TestCompare(true, false, false, &ComputationBuilder::Eq);
62 TestCompare(true, false, true, &ComputationBuilder::Ne);
66 TestCompare(true, false, false, &ComputationBuilder::Le);
70 TestCompare(true, false, false, &ComputationBuilder::Lt);
74 TestCompare(true, false, true, &ComputationBuilder::Ge);
78 TestCompare(true, false, true, &ComputationBuilder
[all...] |
| H A D | scalar_computations_test.cc | 47 ComputationDataHandle (ComputationBuilder::*op)(
51 ComputationBuilder builder(client_, TestName());
60 ComputationDataHandle (ComputationBuilder::*op)(
64 ComputationBuilder builder(client_, TestName());
73 ComputationBuilder builder(client_, TestName());
80 ComputationBuilder builder(client_, TestName());
87 ComputationBuilder builder(client_, TestName());
94 ComputationBuilder builder(client_, TestName());
101 ComputationBuilder builder(client_, TestName());
108 ComputationBuilder builde
[all...] |
| H A D | half_test.cc | 42 std::function<void(ComputationBuilder*, const ComputationDataHandle& src)>;
54 ComputationBuilder builder(client_, TestName());
83 &ComputationBuilder::Abs},
85 &ComputationBuilder::Round},
87 &ComputationBuilder::Ceil},
89 &ComputationBuilder::Cos},
91 &ComputationBuilder::Exp},
93 &ComputationBuilder::Floor},
95 &ComputationBuilder::Log},
97 &ComputationBuilder
[all...] |
| H A D | constants_test.cc | 42 ComputationBuilder builder(client_, TestName());
51 ComputationBuilder builder(client_, TestName());
60 ComputationBuilder builder(client_, TestName());
69 ComputationBuilder builder(client_, TestName());
78 ComputationBuilder builder(client_, TestName());
88 ComputationBuilder builder(client_, TestName());
95 ComputationBuilder builder(client_, TestName());
105 ComputationBuilder builder(client_, TestName());
112 ComputationBuilder builder(client_, TestName());
120 ComputationBuilder builde
[all...] |
| H A D | call_test.cc | 36 ComputationBuilder builder(client_, "Identity");
44 ComputationBuilder builder(client_, "Addition");
54 ComputationBuilder builder(client_, "Addition");
64 ComputationBuilder builder(client_, "Tuple");
77 ComputationBuilder builder(client_, TestName());
86 ComputationBuilder builder(client_, TestName());
96 ComputationBuilder builder(client_, TestName());
106 ComputationBuilder builder(client_, "inner");
113 ComputationBuilder builder2(client_, "outer");
122 ComputationBuilder builder
[all...] |
| H A D | convert_test.cc | 45 ComputationBuilder builder(client_, TestName());
54 ComputationBuilder builder(client_, TestName());
63 ComputationBuilder builder(client_, TestName());
72 ComputationBuilder builder(client_, TestName());
81 ComputationBuilder builder(client_, TestName());
90 ComputationBuilder builder(client_, TestName());
99 ComputationBuilder builder(client_, TestName());
108 ComputationBuilder builder(client_, TestName());
117 ComputationBuilder builder(client_, TestName());
126 ComputationBuilder builde
[all...] |
| H A D | fmax_test.cc | 30 ComputationBuilder builder(client_, TestName());
|
| H A D | vector_ops_simple_test.cc | 52 ComputationBuilder builder(client_, TestName());
66 ComputationBuilder builder(client_, TestName());
87 ComputationBuilder builder(client_, TestName());
109 ComputationBuilder builder(client_, TestName());
120 ComputationBuilder builder(client_, TestName());
129 ComputationBuilder builder(client_, TestName());
139 ComputationBuilder builder(client_, TestName());
150 ComputationBuilder builder(client_, TestName());
162 ComputationBuilder builder(client_, TestName());
170 ComputationBuilder builde
[all...] |
| H A D | convolution_dimension_numbers_test.cc | 42 ComputationBuilder::CreateConvDimensionNumbers(0, 2, 2, 3, 0, 1, 2, 3, 0,
52 ComputationBuilder::CreateConvDimensionNumbers(0, 1, 2, 3, 0, 1, 2, 3, 0,
62 ComputationBuilder::CreateConvDimensionNumbers(0, 1, 2, 3, 0, 2, 2, 3, 0,
79 ComputationBuilder builder(client_, TestName());
86 ComputationBuilder::CreateDefaultConvDimensionNumbers();
|
| H A D | bitcast_convert_test.cc | 45 ComputationBuilder builder(client_, TestName());
54 ComputationBuilder builder(client_, TestName());
63 ComputationBuilder builder(client_, TestName());
75 ComputationBuilder builder(client_, TestName());
84 ComputationBuilder builder(client_, TestName());
93 ComputationBuilder builder(client_, TestName());
103 ComputationBuilder builder(client_, TestName());
115 ComputationBuilder builder(client_, TestName());
132 ComputationBuilder builder(client_, TestName());
|
| H A D | unary_op_test.cc | 40 ComputationBuilder builder(client_, TestName());
53 ComputationBuilder builder(client_, TestName());
62 ComputationBuilder builder(client_, TestName());
72 ComputationBuilder builder(client_, TestName());
89 ComputationBuilder builder(client_, TestName());
105 ComputationBuilder builder(client_, TestName());
117 ComputationBuilder builder(client_, TestName());
142 ComputationBuilder builder(client_, TestName());
158 ComputationBuilder builder(client_, TestName());
190 ComputationBuilder builde
[all...] |
| H A D | binop_scaling_test.cc | 35 ComputationBuilder builder(client_, TestName());
51 ComputationBuilder builder(client_, TestName());
67 ComputationBuilder builder(client_, TestName());
83 ComputationBuilder builder(client_, TestName());
96 ComputationBuilder builder(client_, TestName());
112 ComputationBuilder builder(client_, TestName());
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_helpers.h | 33 static xla::ComputationDataHandle MinValue(xla::ComputationBuilder* b,
38 static xla::ComputationDataHandle MaxValue(xla::ComputationBuilder* b,
43 static xla::ComputationDataHandle Zero(xla::ComputationBuilder* b,
48 static xla::ComputationDataHandle One(xla::ComputationBuilder* b,
53 static xla::ComputationDataHandle Epsilon(xla::ComputationBuilder* b,
59 static xla::ComputationDataHandle IntegerLiteral(xla::ComputationBuilder* b,
65 static xla::ComputationDataHandle FloatLiteral(xla::ComputationBuilder* b,
78 static Status ArgMax(xla::ComputationBuilder* builder,
88 static Status ArgMin(xla::ComputationBuilder* builder,
96 static Status Iota(xla::ComputationBuilder* builde
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| H A D | util.h | 27 xla::ComputationDataHandle Zeros(xla::ComputationBuilder* builder,
32 xla::ComputationDataHandle FloatLiteral(xla::ComputationBuilder* builder,
37 xla::ComputationDataHandle IntegerLiteral(xla::ComputationBuilder* builder,
42 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
48 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
54 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x,
59 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& x);
|
| H A D | cholesky.h | 34 xla::ComputationBuilder* builder, xla::ComputationDataHandle a,
|
| H A D | scatter.h | 48 xla::ComputationBuilder*)>& combiner,
49 xla::ComputationBuilder* builder);
|
| H A D | triangular_solve.h | 61 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& a,
66 xla::ComputationBuilder* builder, const xla::ComputationDataHandle& a,
|
| H A D | while_loop.h | 33 gtl::ArraySlice<xla::ComputationDataHandle>, xla::ComputationBuilder*)>
39 gtl::ArraySlice<xla::ComputationDataHandle>, xla::ComputationBuilder*)>
54 StringPiece name, xla::ComputationBuilder* builder);
63 xla::ComputationBuilder*)>
70 StringPiece name, xla::ComputationBuilder* builder);
|
| H A D | batch_dot.h | 47 xla::ComputationBuilder* builder, xla::ComputationDataHandle x,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | reduction_ops.h | 31 // xla::ComputationBuilder and BuildFinalizer adds the
33 // xla::ComputationBuilder.
41 xla::ComputationBuilder* builder);
47 virtual void BuildReducer(xla::ComputationBuilder* builder,
57 xla::ComputationBuilder* builder,
|
| H A D | reduction_ops.cc | 31 void BuildReducer(xla::ComputationBuilder* builder,
45 xla::ComputationBuilder* builder) override {
49 void BuildReducer(xla::ComputationBuilder* builder,
64 xla::ComputationBuilder* builder) override {
70 void BuildReducer(xla::ComputationBuilder* builder,
84 xla::ComputationBuilder* builder) override {
90 void BuildReducer(xla::ComputationBuilder* builder,
103 void BuildReducer(xla::ComputationBuilder* builder,
110 xla::ComputationBuilder* builder,
127 xla::ComputationBuilder* builde
[all...] |
| H A D | gather_op_helpers.h | 41 xla::ComputationBuilder* builder,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | computation_builder.cc | 37 ComputationBuilder::ComputationBuilder(Client* client, function in class:xla::ComputationBuilder
41 ComputationBuilder::~ComputationBuilder() {}
43 void ComputationBuilder::NoteError(const Status& error) {
54 std::unique_ptr<ComputationBuilder> ComputationBuilder::CreateSubBuilder(
56 auto sub_builder = MakeUnique<ComputationBuilder>(client_, computation_name);
62 Status ComputationBuilder::PrepareComputation() {
86 Status ComputationBuilder
[all...] |