Home
last modified time | relevance | path

Searched refs:SetOpMetadata (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_metadata_test.cc45 builder.SetOpMetadata(metadata_); in TEST_F()
68 builder.SetOpMetadata(metadata_); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.cc111 b->SetOpMetadata(metadata); in Compute()
Dxla_compiler.cc184 builder->SetOpMetadata(retval_metadata); in BuildComputation()
1049 builder->SetOpMetadata(arg_metadata); in BuildArguments()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc404 .def("set_op_metadata", &XlaBuilder::SetOpMetadata) in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h174 void SetOpMetadata(OpMetadata metadata) { metadata_ = std::move(metadata); } in SetOpMetadata() function