Searched refs:mutable_parameters (Results 1 – 5 of 5) sorted by relevance
292 Shape* mutable_parameters(int index) { return ¶meters_.at(index); } in mutable_parameters() function299 std::vector<Shape>* mutable_parameters() { return ¶meters_; } in mutable_parameters() function
169 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in SetToDefaultLayout()269 for (auto& parameter_shape : *program_shape->mutable_parameters()) { in ClearLayout()
82 arg_layouts[i] = xla::Shape(*pshape->mutable_parameters(i)); in CompileXla()
3705 new_program_shape.mutable_parameters(i))); in Run()
300 *program_shape.mutable_parameters(index) = Shape(instr.shape()); in GetProgramShape()