Searched refs:sas_builder (Results 1 – 1 of 1) sorted by relevance
445 NodeDefBuilder sas_builder(sas_name, "_ScopedAllocatorSplit"); in BuildSplitNode() local446 sas_builder.Device(device_name); in BuildSplitNode()447 sas_builder.Attr("sa_name", sa_name); in BuildSplitNode()448 sas_builder.Attr("id", sa_id); in BuildSplitNode()449 sas_builder.Attr("T", dtype); in BuildSplitNode()450 sas_builder.Attr("shapes", input_shapes); in BuildSplitNode()452 sas_builder.Attr("N", static_cast<int>(sas_inputs.size())); in BuildSplitNode()453 sas_builder.Input(NodeDefBuilder::NodeOut(sa_op_name, 0, dtype)); in BuildSplitNode()454 sas_builder.Input(sas_inputs); in BuildSplitNode()456 LOG_WARNING_AND_RETURN_IF_ERROR(sas_builder.Finalize(sas_node)); in BuildSplitNode()