| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 28 final_op='Id', argument
68 final_op=final_op,
79 final_op='Id', argument
117 final_op=final_op,
|
| D | collective_ops_test.py | 57 final_op='Div', argument
81 final_op,
290 final_op='Id')
301 final_op='Id')
468 merge_op='Add', final_op='Id')
473 merge_op='Add', final_op='Id')
|
| D | collective_ops_gpu_test.py | 261 final_op='Id', communication_hint='nccl')
266 final_op='Id', communication_hint='nccl')
285 final_op='Id', communication_hint='NCCL')
|
| D | op_selector.py | 353 final_op = _as_operation(tensor)
354 while path_op != final_op:
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 31 if (col_params_->final_op) { in Run()
82 if (col_params_->final_op) { in Run()
97 if (!col_params_->final_op) { in Run()
119 col_params_->final_op, col_ctx_->output, &group_size); in Run()
|
| D | collective_ops.cc | 270 col_params_->final_op = final_op_.get(); in CollectiveReduceOpKernel()
536 col_params->final_op = final_op_.get(); in ComputeAsync()
|
| D | collective_nccl_test.cc | 475 di->col_params_->final_op = di->final_op_.get(); in InitDevice()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 106 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv() local
109 const auto* bts_op = has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv()
117 Operator* bias_add_op = !has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv()
146 bias_add_op->outputs[0] = final_op->outputs[0]; in ResolveDilatedConv()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveReduce.pbtxt | 49 name: "final_op"
112 name: "final_op"
183 name: "final_op"
261 name: "final_op"
|
| D | CollectiveReduceV2.pbtxt | 49 name: "final_op"
115 name: "final_op"
193 name: "final_op"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveReduce.pbtxt | 49 name: "final_op"
112 name: "final_op"
183 name: "final_op"
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/surface/ |
| D | call.cc | 250 } final_op; member
364 call->final_op.client.status_details = nullptr; in grpc_call_create()
365 call->final_op.client.status = nullptr; in grpc_call_create()
366 call->final_op.client.error_string = nullptr; in grpc_call_create()
382 call->final_op.server.cancelled = nullptr; in grpc_call_create()
383 call->final_op.server.core_server = args->server; in grpc_call_create()
485 } else if (call->final_op.server.core_server != nullptr) { in grpc_call_create()
487 call->final_op.server.core_server->channelz_node(); in grpc_call_create()
751 call->final_op.client.status, in set_final_status()
752 call->final_op.client.status_details, nullptr, in set_final_status()
[all …]
|
| /external/grpc-grpc/src/core/lib/surface/ |
| D | call.cc | 209 } final_op; member
335 call->final_op.server.server = args->server; in grpc_call_create()
447 grpc_server_get_channelz_node(call->final_op.server.server); in grpc_call_create()
705 call->final_op.client.status, in set_final_status()
706 call->final_op.client.status_details, nullptr, in set_final_status()
707 call->final_op.client.error_string); in set_final_status()
709 grpc_slice_ref_internal(*call->final_op.client.status_details); in set_final_status()
714 if (*call->final_op.client.status != GRPC_STATUS_OK) { in set_final_status()
721 *call->final_op.server.cancelled = in set_final_status()
726 grpc_server_get_channelz_node(call->final_op.server.server); in set_final_status()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_reducer.cc | 123 if (col_params_->final_op) { in ContinueAfterInputCopy()
269 if (!rf->second_pass && col_params_->final_op && rf->is_final) { in RunAsyncParts()
274 col_params_->final_op, &rf->chunk, &group_size_tensor_); in RunAsyncParts()
|
| D | ring_reducer_test.cc | 480 col_params_->final_op = final_op_.get(); in DoReduce()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 400 const std::string final_op("Id"); in CollectiveSum() local
401 TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(), in CollectiveSum()
402 final_op.length()); in CollectiveSum()
491 const std::string final_op("Id"); in RegisterCollectiveMulFunction() local
492 TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(), in RegisterCollectiveMulFunction()
493 final_op.length()); in RegisterCollectiveMulFunction()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 119 final_op="Id",
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 147 OpKernel* final_op = nullptr; // reduction only member
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 68 final_op="Id")
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 467 " instance_key=instance_key, merge_op='Add', final_op='Div',\n",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 781 …put\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offse…
785 …', \'group_key\', \'instance_key\', \'ordering_token\', \'merge_op\', \'final_op\', \'communicatio…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 781 …put\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offse…
785 …', \'group_key\', \'instance_key\', \'ordering_token\', \'merge_op\', \'final_op\', \'communicatio…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 19302 …t, instance_key tf.Output, ordering_token []tf.Output, merge_op string, final_op string, optional …
19306 attrs := map[string]interface{}{"merge_op": merge_op, "final_op": final_op}
43228 …group_size int64, group_key int64, instance_key int64, merge_op string, final_op string, subdiv_of…
43232 …up_key, "instance_key": instance_key, "merge_op": merge_op, "final_op": final_op, "subdiv_offsets"…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1905 TF_AnyStrAttrOf<["Id", "Div"]>:$final_op,
1932 TF_AnyStrAttrOf<["Id", "Div"]>:$final_op,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7826 name: "final_op"
7916 name: "final_op"
|