Home
last modified time | relevance | path

Searched refs:use_spmd_for_xla_partitioning (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_rewrite.mlir8 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
23 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
39 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
54 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
69 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
84 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
99 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
114 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
129 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
144 …rding_configuration = [], output_sharding_configuration = [], use_spmd_for_xla_partitioning = fals…
[all …]
Dtpu_resource_partitioning.mlir16 …%2 = "tf_device.cluster_func"(%1) {func = @computation, use_spmd_for_xla_partitioning = true} : (t…
37 …%2 = "tf_device.cluster_func"(%1) {func = @computation, use_spmd_for_xla_partitioning = true} : (t…
55 …%5 = "tf_device.cluster_func"(%4) {func = @computation, use_spmd_for_xla_partitioning = false} : (…
113 …%2 = "tf_device.cluster_func"(%1) {func = @computation, use_spmd_for_xla_partitioning = true} : (t…
Dtpu_sharding_identification.mlir268 …%1:2 = "tf_device.cluster_func"(%0, %arg1) {func = @cluster_func, use_spmd_for_xla_partitioning = …
291 …"tf_device.cluster_func"(%1) {func = @cluster_func, use_spmd_for_xla_partitioning = true} : (tenso…
312 …%1 = "tf_device.cluster_func"(%0) {func = @cluster_func, use_spmd_for_xla_partitioning = true} : (…
335 …%1 = "tf_device.cluster_func"() {func = @cluster_func, use_spmd_for_xla_partitioning = true} : () …
Dgraph_pruning_preserve_ops.mlir12 …, step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = true…
Dtpu_cluster_formation.mlir394 …, step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = fals…
419 …, step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = fals…
Dtpu_space_to_depth_pass.mlir106 …, step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = fals…
180 …, step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = fals…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.tpu.-x-l-a-options.pbtxt7 name: "use_spmd_for_xla_partitioning"
Dtensorflow.raw_ops.pbtxt4817 …_map\', \'step_marker_location\', \'allow_soft_placement\', \'use_spmd_for_xla_partitioning\', \'n…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUReplicateMetadata.pbtxt280 name: "use_spmd_for_xla_partitioning"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir12 …cation = "STEP_MARK_AT_ENTRY", topology = "", use_tpu = true, use_spmd_for_xla_partitioning = fals…
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dcompile_metadata.proto117 bool use_spmd_for_xla_partitioning = 15; field
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py876 def __new__(cls, use_spmd_for_xla_partitioning=True): argument
877 return super(XLAOptions, cls).__new__(cls, use_spmd_for_xla_partitioning)
1372 xla_options.use_spmd_for_xla_partitioning
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py1033 xla_options=tpu.XLAOptions(use_spmd_for_xla_partitioning=self
1489 xla_options=tpu.XLAOptions(use_spmd_for_xla_partitioning=self
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_passes.td563 …ation = "tf_device.cluster_func"(%read) {func = @computation, use_spmd_for_xla_partitioning = true…
580 …evice.cluster_func"(%partitioned_input) {func = @computation, use_spmd_for_xla_partitioning = true…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td842 DefaultValuedAttr<BoolAttr, "false">:$use_spmd_for_xla_partitioning
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4817 …_map\', \'step_marker_location\', \'allow_soft_placement\', \'use_spmd_for_xla_partitioning\', \'n…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt54039 name: "use_spmd_for_xla_partitioning"