Searched refs:pack_builder (Results 1 – 2 of 2) sorted by relevance
124 NodeDefBuilder pack_builder( in ReplicateFromCompositeDeviceToRegularDevice() local128 pack_builder.Attr("N", num_replicas); in ReplicateFromCompositeDeviceToRegularDevice()130 pack_builder.Attr("T", dtype); in ReplicateFromCompositeDeviceToRegularDevice()137 pack_builder.Input(inputs); in ReplicateFromCompositeDeviceToRegularDevice()139 TF_RETURN_IF_ERROR(pack_builder.Finalize(&pack_def)); in ReplicateFromCompositeDeviceToRegularDevice()
26 "pack_builder.cc",59 "pack_builder.h",