Searched refs:is_mirrored_variable (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TPUReplicatedInput.pbtxt | 45 name: "is_mirrored_variable"
74 name: "is_mirrored_variable"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUReplicatedInput.pbtxt | 45 name: "is_mirrored_variable"
81 name: "is_mirrored_variable"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | graph_pruning_preserve_ops.mlir | 40 …r.island(%1) wraps "tf.TPUReplicatedInput"(%2) {index = -1 : i64, is_mirrored_variable = false, is…
|
| D | tpu_cluster_formation.mlir | 360 // Test that the `is_mirrored_variable` attribute is preserved in the
366 …%1 = "tf.TPUReplicatedInput"(%arg2, %arg3) {index = 1 : i64, is_mirrored_variable = true} : (tenso…
367 …%2 = "tf.TPUReplicatedInput"(%arg4) {index = 2 : i64, is_mirrored_variable = true, is_packed = tru…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
| D | executor_tpuv1_island_coarsening.mlir | 102 …nput"(%arg0) {N = 1 : i64, T = i32, device = "", index = 0 : i64, is_mirrored_variable = false} : …
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 415 if (tpu_replicated_input.is_mirrored_variable()) in ReplicateCluster()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 2635 bool is_mirrored_variable; in BuildTPUReplicateOps() local
2637 &is_mirrored_variable) in BuildTPUReplicateOps()
2639 if (is_mirrored_variable) { in BuildTPUReplicateOps()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 376 is_mirrored_variable=is_mirrored,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4821 …argspec: "args=[\'inputs\', \'is_mirrored_variable\', \'index\', \'is_packed\', \'name\'], varargs…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4821 …argspec: "args=[\'inputs\', \'is_mirrored_variable\', \'index\', \'is_packed\', \'name\'], varargs…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 15825 DefaultValuedAttr<BoolAttr, "false">:$is_mirrored_variable,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 54068 name: "is_mirrored_variable"
|