Searched refs:SetOpMetadata (Results 1 – 5 of 5) sorted by relevance
45 builder.SetOpMetadata(metadata_); in TEST_F()68 builder.SetOpMetadata(metadata_); in TEST_F()
111 b->SetOpMetadata(metadata); in Compute()
184 builder->SetOpMetadata(retval_metadata); in BuildComputation()1049 builder->SetOpMetadata(arg_metadata); in BuildArguments()
404 .def("set_op_metadata", &XlaBuilder::SetOpMetadata) in BuildXlaCompilerSubmodule()
174 void SetOpMetadata(OpMetadata metadata) { metadata_ = std::move(metadata); } in SetOpMetadata() function