Searched defs:result_layout (Results 1 – 13 of 13) sorted by relevance
36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout()67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout() function
45 const Shape* result_layout = nullptr; member
744 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints() local1101 const ShapeLayout& result_layout = in CheckLayouts() local2209 ShapeLayout* result_layout = computation_layout->mutable_result_layout(); in PropagateComputationLayouts() local
1617 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout() local2256 const Layout& result_layout = result_shape.layout(); in Postprocess() local
2643 auto result_layout = program_layout.result_layout().layout(); local
58 const Shape* ExecutableBuildOptions::result_layout() const { in result_layout() function in xla::ExecutableBuildOptions93 string result_layout = "nullopt"; in ToString() local
46 const Shape* result_layout; member
70 LayoutProto result_layout = 2; field
60 absl::optional<const Shape> result_layout, in CreateModuleConfig()139 absl::optional<const Shape> result_layout, in CreateModuleConfig()
197 Shape result_layout; in DetermineArgumentLayoutsFromCompileOptions() local
403 xla::ShapeLayout result_layout( in ConfigWithLayout() local
276 XLA_Shape result_layout; member
1232 xla::LayoutProto result_layout = in ExportXlaOp() local