Searched refs:SetOpMetadata (Results 1 – 9 of 9) sorted by relevance
43 builder.SetOpMetadata(metadata_); in TEST_F()64 builder.SetOpMetadata(metadata_); in TEST_F()
98 b->SetOpMetadata(metadata); in Compute()
175 builder->SetOpMetadata(retval_metadata); in BuildComputation()785 builder->SetOpMetadata(arg_metadata); in BuildArguments()
416 void ComputationBuilder::SetOpMetadata(const OpMetadata& metadata) { in SetOpMetadata() function in xla::swig::ComputationBuilder417 builder_.SetOpMetadata(metadata); in SetOpMetadata()
210 void SetOpMetadata(const OpMetadata& metadata);
288 %unignore xla::swig::ComputationBuilder::SetOpMetadata;
934 def SetOpMetadata(self, op_metadata): member in ComputationBuilder936 self._client.SetOpMetadata(op_metadata)
1704 c.SetOpMetadata(xla_client.CurrentSourceInfoMetadata())
153 void SetOpMetadata(const OpMetadata& metadata) { metadata_ = metadata; } in SetOpMetadata() function