Searched refs:add_parameter_layout (Results 1 – 5 of 5) sorted by relevance
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout() function
171 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()
794 computation_layout->add_parameter_layout(ShapeLayout(param)); in ParseComputationLayout()
460 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
447 layout.add_parameter_layout(xla::ShapeLayout( in ConfigWithLayout()