- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for add_tile_assignment_devices (0.18 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} void SetMaximalSharding(NodeDefBuilder& node_builder) { xla::OpSharding sharding; sharding.set_type(xla::OpSharding::MAXIMAL); sharding.add_tile_assignment_dimensions(1); sharding.add_tile_assignment_devices(0); node_builder.Attr("_XlaSharding", sharding.SerializeAsString()); } // Builds XlaSendToHost node which sends cond predicate to host. TF_ATTRIBUTE_NOINLINE absl::StatusOr<Node*> BuildSendIfPredNode(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)