Searched refs:ComputationBuilder (Results 1 – 5 of 5) sorted by relevance
284 %unignore xla::swig::ComputationBuilder;285 %unignore xla::swig::ComputationBuilder::ComputationBuilder;286 %unignore xla::swig::ComputationBuilder::Build;287 %unignore xla::swig::ComputationBuilder::BuildWithRoot;288 %unignore xla::swig::ComputationBuilder::SetOpMetadata;289 %unignore xla::swig::ComputationBuilder::ClearOpMetadata;290 %unignore xla::swig::ComputationBuilder::Parameter;291 %unignore xla::swig::ComputationBuilder::GetShape;292 %unignore xla::swig::ComputationBuilder::GetReturnValueShape;293 %unignore xla::swig::ComputationBuilder::ReplicaId;[all …]
413 ComputationBuilder::ComputationBuilder(const string& computation_name) in ComputationBuilder() function in xla::swig::ComputationBuilder416 void ComputationBuilder::SetOpMetadata(const OpMetadata& metadata) { in SetOpMetadata()420 void ComputationBuilder::ClearOpMetadata() { builder_.ClearOpMetadata(); } in ClearOpMetadata()422 StatusOr<Computation*> ComputationBuilder::Build() { in Build()427 LocalOp ComputationBuilder::Parameter(int64 parameter_number, in Parameter()432 StatusOr<Computation*> ComputationBuilder::BuildWithRoot(const LocalOp& root) { in BuildWithRoot()437 StatusOr<Shape> ComputationBuilder::GetShape(const LocalOp& operand) { in GetShape()441 StatusOr<Shape> ComputationBuilder::GetReturnValueShape() { in GetReturnValueShape()446 LocalOp ComputationBuilder::ReplicaId() { return xla::ReplicaId(&builder_); } in ReplicaId()448 LocalOp ComputationBuilder::Infeed(const Shape& shape) { in Infeed()[all …]
899 class ComputationBuilder(object): class915 self._client = c_api.ComputationBuilder(name.encode('utf8'))1812 getattr(c_api.ComputationBuilder, method_name))1814 setattr(ComputationBuilder, method_name, forward)1818 getattr(c_api.ComputationBuilder, method_name), is_binop=True)1820 setattr(ComputationBuilder, method_name, forward)
206 class ComputationBuilder {208 ComputationBuilder(const string& computation_name);
51 return xla_client.ComputationBuilder(name)104 builder = xla_client.ComputationBuilder("acomputation")