Home
last modified time | relevance | path

Searched refs:ParseShardingFromEdgeSource (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.h47 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromEdgeSource(
Dsharding_util.cc115 xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromEdgeSource( in ParseShardingFromEdgeSource() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2124 ParseShardingFromEdgeSource(*edge, num_cores_per_replica, in AssignArgsAndRetvalsToCores()