| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute.py | 69 def __init__(self, input_dataset, num_workers, index, num_replicas=None): argument
88 def _AutoShardDatasetV1(input_dataset, num_workers, index, num_replicas=None): # pylint: disable=i… argument
244 def __init__(self, input_dataset, num_replicas): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op.cc | 47 int64 index, num_workers, auto_shard_policy, num_replicas; in MakeDataset() local
75 int64 num_replicas) { in CreateConfig()
|
| D | rebatch_dataset_op.cc | 44 int64 num_replicas; in MakeDataset() local
58 const int64 num_replicas, const DataTypeVector& output_types, in Dataset()
104 Node* num_replicas = nullptr; in AsGraphDefInternal() local
|
| D | auto_shard_dataset_op_test.cc | 28 int auto_shard_policy, int64 num_replicas, in AutoShardDatasetParams()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_module_loader.h | 30 int64 num_replicas = 1; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_invariant_op_hoisting.cc | 76 void MakeShapeOpInvariant(tf_device::ReplicateOp replicate_op, int num_replicas, in MakeShapeOpInvariant()
155 const int num_replicas = replicate_op.n(); in HoistReplicateInvariantOps() local
|
| D | tpu_variable_runtime_reformatting.cc | 180 int64_t num_replicas = replicate.n(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() local
275 int64_t num_replicas = replicate.n(); in AddInputsToReplicateOp() local
384 int64_t num_replicas = replicate.n(); in HandleReplicateOp() local
|
| D | replicate_to_island.cc | 141 int num_replicas, llvm::SmallVectorImpl<tf_executor::IslandOp>& replicas) { in ExpandReplicateIntoReplicas()
239 const int num_replicas = replicate_op.n(); in CreateIslandsFromReplicate() local
|
| D | tpu_rewrite_pass.cc | 303 tf_device::ClusterFuncOp op, int num_replicas, int num_cores_per_replica, in SetMetadataProtoFromClusterFuncOp()
350 tf_device::ClusterFuncOp cluster_func, int num_replicas, in BuildCompileOp()
418 const int num_replicas = tpu_devices.size(); in AssignDevicesToReplicate() local
571 int num_replicas = 1; in Rewrite() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_parallel.h | 29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) { in AutoParallel()
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_test_util.py | 51 num_replicas=1): argument
|
| D | resnet50_test.py | 37 def compute_gradients(model, images, labels, num_replicas=1): argument
268 num_replicas=1): argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 59 [](PyTpuClient* client, int num_replicas) in PYBIND11_MODULE()
60 -> StatusOr<std::vector<std::shared_ptr<PjRtDevice>>> { in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 55 const xrt::DeviceAssignment& xrt_device_assignment, int num_replicas, in GenerateXlaDeviceAssignment()
132 int num_replicas = config.num_replicas() ? config.num_replicas() : 1; in Compile() local
|
| D | tpu_compile_ops.cc | 92 tensorflow::tpu::TpuMeshStateInterface* mesh_state, int num_replicas, in CompilationCacheKey()
177 int num_replicas = config.num_replicas() ? config.num_replicas() : 1; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner_interface.h | 47 int64 num_replicas = 1; member
|
| D | ar_crs_combiner.h | 75 ArCrsCombiner(int num_spatial_partitions, int num_replicas, in ArCrsCombiner()
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 55 def _collective_reduce(inputs, operation, num_replicas): argument
73 def _collective_sum(inputs, num_replicas): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 63 int num_replicas) { in set_num_replicas()
|
| D | executable_build_options.h | 80 int num_replicas() const { return num_replicas_; } in num_replicas() function
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 207 int64 num_replicas, bool use_threads, bool run_hlo_passes) { in ExecuteReplicated()
220 int64 num_replicas, DeviceAssignment* device_assignment, in ExecuteReplicated()
237 int64 num_replicas, bool run_hlo_passes) { in ExecuteReplicated()
419 int64 num_replicas, in RunReplicated()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.cc | 69 PyClient::GetDefaultDeviceAssignment(int num_replicas, int num_partitions) { in GetDefaultDeviceAssignment()
88 PyClient::GetDefaultDeviceAssignment1D(int num_replicas) { in GetDefaultDeviceAssignment1D()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | compile_metadata.proto | 80 int32 num_replicas = 3; field
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 562 Status RewriteRebatchV2ToV1(const NodeDef& sink_node, int64 num_replicas, in RewriteRebatchV2ToV1()
612 int64 num_replicas, MutableGraphView* graph) { in ShardByData()
628 AutoShardPolicy policy, int64 num_replicas, in OptimizeGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 188 int num_replicas, int num_cores_per_replica, in GetFullMeshTPUExecutionDeviceAssignment()
347 int num_replicas, int num_cores_per_replica, in GetGeneralTPUExecutionDeviceAssignment()
450 Devices devices, int num_replicas, int num_cores_per_replica, in GetTPUCompilationAndExecutionDevices()
|