Searched refs:max_subdivs_per_device (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer_test.cc | 274 cp->instance.impl_details.max_subdivs_per_device = 0; in TEST_F()
310 cp->instance.impl_details.max_subdivs_per_device = 0; in TEST_F()
328 cp->instance.impl_details.max_subdivs_per_device = 4; in TEST_F()
335 cp->instance.impl_details.max_subdivs_per_device = 4; in TEST_F()
355 cp->instance.impl_details.max_subdivs_per_device = 0; in TEST_F()
373 cp->instance.impl_details.max_subdivs_per_device = -1; in TEST_F()
|
| D | ring_alg.cc | 115 if (col_params->instance.impl_details.max_subdivs_per_device == -1) { in GenerateSubdivsInCollectiveParams()
126 const int max_subdivs_per_device = in GenerateSubdivsInCollectiveParams() local
127 (col_params->instance.impl_details.max_subdivs_per_device > 0) in GenerateSubdivsInCollectiveParams()
128 ? col_params->instance.impl_details.max_subdivs_per_device in GenerateSubdivsInCollectiveParams()
130 const int kMaxNumSubdivs = max_subdivs_per_device * kAvgDevPerTask; in GenerateSubdivsInCollectiveParams()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 104 max_subdivs_per_device=-1, argument
155 max_subdivs_per_device=max_subdivs_per_device,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveReduceV2.pbtxt | 311 name: "max_subdivs_per_device"
405 name: "max_subdivs_per_device"
499 name: "max_subdivs_per_device"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 485 max_subdivs_per_device=[-1, 0, 16]), device_combination))
493 max_subdivs_per_device): argument
509 if max_subdivs_per_device == -1:
525 max_subdivs_per_device=max_subdivs_per_device)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 109 int max_subdivs_per_device = -1; // Upper bound on subdivisions per device. member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.td | 619 $max_subdivs_per_device
624 $max_subdivs_per_device
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 837 …\', \'final_op\', \'communication_hint\', \'timeout_seconds\', \'max_subdivs_per_device\', \'name\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 837 …\', \'final_op\', \'communication_hint\', \'timeout_seconds\', \'max_subdivs_per_device\', \'name\…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 8753 name: "max_subdivs_per_device"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2317 DefaultValuedAttr<I64Attr, "-1">:$max_subdivs_per_device
|