Searched refs:XlaOutputDescription (Results 1 – 4 of 4) sorted by relevance
25 // CHECK-NEXT: // XlaOutputDescription type=float shape=()40 // TUPLE-ARGS-NEXT: // XlaOutputDescription type=float shape=()53 // NO_RET_TUPLE-NEXT: // XlaOutputDescription type=float shape=()
98 struct XlaOutputDescription { struct154 std::vector<XlaOutputDescription> outputs;
135 using OutputDescription = ::tensorflow::XlaOutputDescription;
139 xla::Shape* xla_output_shape, std::vector<XlaOutputDescription>* outputs, in GetOutputInfo()183 XlaOutputDescription& out_desc = outputs->back(); in GetOutputInfo()