Searched refs:shape_with_output_layout (Results 1 – 5 of 5) sorted by relevance
114 const Shape* shape_with_output_layout) { in ExecuteAndTransfer() argument116 if (shape_with_output_layout != nullptr) { in ExecuteAndTransfer()118 shape_with_output_layout->ToProto(); in ExecuteAndTransfer()126 const Shape* shape_with_output_layout) { in ExecuteAndTransfer() argument129 return ExecuteAndTransfer(computation, arguments, shape_with_output_layout); in ExecuteAndTransfer()134 const Shape* shape_with_output_layout) { in ExecuteAndTransferReference() argument136 if (shape_with_output_layout != nullptr) { in ExecuteAndTransferReference()138 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);
174 absl::optional<Shape> shape_with_output_layout; in ExecuteAndTransfer() local176 shape_with_output_layout = in ExecuteAndTransfer()177 Shape(execution_options->shape_with_output_layout()); in ExecuteAndTransfer()179 return Transfer(*data, shape_with_output_layout.has_value() in ExecuteAndTransfer()180 ? &(*shape_with_output_layout) in ExecuteAndTransfer()
293 const Shape shape_with_output_layout( in CreateModuleConfig() local294 execution_options->shape_with_output_layout()); in CreateModuleConfig()296 ValidateResultShape(shape_with_output_layout, program_shape.result())); in CreateModuleConfig()299 shape_with_output_layout)); in CreateModuleConfig()
284 ShapeProto shape_with_output_layout = 2; field