Searched defs:ComputationBuilder (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
H A Dcomputation_builder.h51 class ComputationBuilder { class in namespace:xla
55 ComputationBuilder(Client* client, const string& computation_name);
57 ~ComputationBuilder();
821 // Returns a new ComputationBuilder whose resultant Computation is used only
822 // by this ComputationBuilder. The sub-ComputationBuilder has the same
824 std::unique_ptr<ComputationBuilder> CreateSubBuilder(
829 // last expression enqueued on the ComputationBuilder. Any subsequent
830 // operations added to the ComputationBuilder will not have any effect unless
839 // the parent ComputationBuilder an
[all...]
H A Dcomputation_builder.cc37 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...]
/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client.py34 # whereas method names of ComputationBuilder and LocalComputation are
354 ComputationBuilder methods.
443 class ComputationBuilder(object): class in inherits:object
1087 """Forward remaining ComputationBuilder methods to the C API.
1114 setattr(ComputationBuilder, method_name, forward)
1120 setattr(ComputationBuilder, method_name, forward)

Completed in 262 milliseconds