Searched refs:shape_with_output_layout (Results 1 – 5 of 5) sorted by relevance
75 const Shape shape_with_output_layout( in CreateModuleConfig() local76 execution_options->shape_with_output_layout()); in CreateModuleConfig()78 ValidateResultShape(shape_with_output_layout, program_shape.result())); in CreateModuleConfig()81 shape_with_output_layout)); in CreateModuleConfig()
113 const Shape* shape_with_output_layout) { in ExecuteAndTransfer() argument115 if (shape_with_output_layout != nullptr) { in ExecuteAndTransfer()117 shape_with_output_layout->ToProto(); in ExecuteAndTransfer()125 const Shape* shape_with_output_layout) { in ExecuteAndTransfer() argument128 return ExecuteAndTransfer(computation, arguments, shape_with_output_layout); in ExecuteAndTransfer()133 const Shape* shape_with_output_layout) { in ExecuteAndTransferReference() argument135 if (shape_with_output_layout != nullptr) { in ExecuteAndTransferReference()137 shape_with_output_layout->ToProto(); in ExecuteAndTransferReference()
100 const Shape* shape_with_output_layout = nullptr);105 const Shape* shape_with_output_layout = nullptr);113 const Shape* shape_with_output_layout = nullptr);
173 std::optional<Shape> shape_with_output_layout; in ExecuteAndTransfer() local175 shape_with_output_layout = in ExecuteAndTransfer()176 Shape(execution_options->shape_with_output_layout()); in ExecuteAndTransfer()178 return Transfer(*data, shape_with_output_layout.has_value() in ExecuteAndTransfer()179 ? &(*shape_with_output_layout) in ExecuteAndTransfer()
455 ShapeProto shape_with_output_layout = 2; field