Searched refs:mutable_parameters (Results 1 – 4 of 4) sorted by relevance
241 Shape* mutable_parameters(int index) { return ¶meters_.at(index); } in mutable_parameters() function248 std::vector<Shape>* mutable_parameters() { return ¶meters_; } in mutable_parameters() function
167 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in SetToDefaultLayout()267 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in ClearLayout()
68 arg_layouts[i] = xla::Shape(*pshape->mutable_parameters(i)); in CompileXla()
178 *program_shape.mutable_parameters(index) = Shape(instr.shape()); in GetProgramShape()