| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 23 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
58 final_op=final_op,
|
| D | collective_ops_test.py | 45 instance_key=1, merge_op='Add', final_op='Div'): argument
59 tensor, group_size, group_key, instance_key, merge_op, final_op,
235 final_op='Id')
245 final_op='Id')
401 merge_op='Add', final_op='Id')
406 merge_op='Add', final_op='Id')
|
| D | op_selector.py | 353 final_op = _as_operation(tensor)
354 while path_op != final_op:
|
| D | collective_ops_gpu_test.py | 275 final_op='Id', communication_hint='nccl')
280 final_op='Id', communication_hint='nccl')
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 60 if (col_params_->final_op) { in Run()
111 if (col_params_->final_op) { in Run()
189 if (!col_params_->final_op) { in Run()
211 col_params_->final_op.get(), col_ctx_->output, &group_size); in Run()
|
| D | collective_ops.cc | 195 col_params_.final_op = BuildOpKernel(c, final_op_name, &sub_node); in CollectiveReduceOpKernel()
|
| D | collective_nccl_test.cc | 463 di->col_params_.final_op = GetDiv(di->device_); 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_v1/ |
| D | CollectiveReduce.pbtxt | 49 name: "final_op"
112 name: "final_op"
183 name: "final_op"
|
| /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 | 124 if (col_params_->final_op) { in ContinueAfterInputCopy()
270 if (!rf->second_pass && col_params_->final_op.get() && in RunAsyncParts()
276 col_params_->final_op.get(), &rf->chunk, &group_size_tensor_); in RunAsyncParts()
|
| D | ring_reducer_test.cc | 476 col_params_.final_op = in DoReduce()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 132 std::unique_ptr<OpKernel> final_op; // reduction only member
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 713 …put\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offse…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 713 …put\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offse…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 47215 …group_size int64, group_key int64, instance_key int64, merge_op string, final_op string, subdiv_of…
47219 …up_key, "instance_key": instance_key, "merge_op": merge_op, "final_op": final_op, "subdiv_offsets"…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 13091 name: "final_op"
13154 name: "final_op"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7191 name: "final_op"
|