Home
last modified time | relevance | path

Searched defs:num_replicated_inputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc344 int32_t num_replicated_inputs = replicated_inputs.size() * n; in ParseReplicateOp() local
381 const int32_t num_replicated_inputs = in Print() local
451 const int32_t num_replicated_inputs = in Verify() local
545 int32_t num_replicated_inputs = replicated_inputs.size() * n; in BuildReplicateOp() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc937 int num_replicated_inputs = num_per_replica_inputs + num_distributed_vars; in RemoveUnusedXlaInput() local
2670 const int num_replicated_inputs = distributed_var_start_index; in BuildTPUReplicateOps() local