Searched refs:xla_output_shape (Results 1 – 4 of 4) sorted by relevance
210 << " -> " << xla::ShapeUtil::HumanString(body.xla_output_shape); in Compile()212 << " -> " << xla::ShapeUtil::HumanString(cond.xla_output_shape); in Compile()221 ctx, xla::ShapeUtil::Compatible(body_input_shape, body.xla_output_shape), in Compile()225 xla::ShapeUtil::HumanString(body.xla_output_shape))); in Compile()230 xla::ShapeUtil::Compatible(cond.xla_output_shape, in Compile()234 xla::ShapeUtil::HumanString(cond.xla_output_shape))); in Compile()
654 result->xla_output_shape.Swap( in CompileGraph()657 << xla::ShapeUtil::HumanString(result->xla_output_shape); in CompileGraph()660 xla::LayoutUtil::SetToDefaultLayout(&result->xla_output_shape); in CompileGraph()673 xla::ShapeUtil::GetTupleElementShape(result->xla_output_shape, in CompileGraph()
212 xla::Shape xla_output_shape; member
222 build_options.set_result_layout(result.xla_output_shape); in BuildExecutable()