Searched defs:XlaBuilder (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 179 [](xla::XlaOp x, xla::XlaOp y, xla::XlaBuilder*) { in Compile()
192 [](xla::XlaOp x, xla::XlaOp y, xla::XlaBuilder*) { in Compile()
205 [](xla::XlaOp x, xla::XlaOp y, xla::XlaBuilder*) { in Compile()
218 [](xla::XlaOp x, xla::XlaOp y, xla::XlaBuilder*) { in Compile()
231 context, [](xla::XlaOp, xla::XlaOp y, xla::XlaBuilder*) { return y; }); in Compile()
|
| D | reduction_ops_common.cc | 41 xla::XlaBuilder* /*builder*/, const xla::XlaOp& /*input*/, in BuildFinalizer()
|
| D | reduction_ops.cc | 131 xla::XlaBuilder* /*builder*/, const xla::XlaOp& input, in BuildFinalizer()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_computation.h | 60 friend class XlaBuilder; variable
|
| D | xla_builder.h | 48 class XlaBuilder; variable
116 friend class XlaBuilder; variable
|
| D | xla_builder.cc | 243 XlaBuilder::XlaBuilder(const string& computation_name) in XlaBuilder() function in xla::XlaBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 420 XlaBuilder = _xla.XlaBuilder variable
|