Searched refs:add_parameter_layout (Results 1 – 4 of 4) sorted by relevance
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout() function
137 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()
456 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
407 layout.add_parameter_layout(xla::ShapeLayout( in ConfigWithLayout()