Home
last modified time | relevance | path

Searched defs:XlaCompilationResult (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h130 struct XlaCompilationResult { struct
135 std::vector<int> input_mapping;
139 std::vector<xla::Shape> xla_input_shapes;
143 xla::Shape xla_output_shape;
148 std::vector<XlaOutputDescription> outputs;
152 tf2xla::HostComputeMetadata host_compute_metadata;
158 std::vector<XlaResourceUpdate> resource_updates;
161 std::shared_ptr<xla::XlaComputation> computation;
164 struct CollectiveInfo {
195 const XlaCompilationResult::CollectiveInfo& collective_info, argument