Searched refs:SetOpMetadata (Results 1 – 7 of 7) sorted by relevance
43 builder.SetOpMetadata(metadata_); in TEST_F()65 builder.SetOpMetadata(metadata_); in TEST_F()
94 b->SetOpMetadata(metadata); in Compute()
176 builder->SetOpMetadata(retval_metadata); in BuildComputation()1013 builder->SetOpMetadata(arg_metadata); in BuildArguments()
740 def SetOpMetadata(self, op_metadata): member in ComputationBuilder742 self._builder.SetOpMetadata(op_metadata)
805 .def("SetOpMetadata", &XlaBuilder::SetOpMetadata) in PYBIND11_MODULE()
1990 c.SetOpMetadata(xla_client.CurrentSourceInfoMetadata())2008 c.SetOpMetadata(xla_client.CurrentSourceInfoMetadata())
153 void SetOpMetadata(OpMetadata metadata) { metadata_ = std::move(metadata); } in SetOpMetadata() function