| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_tensor_handle.h | 27 : op_id(tensor_handle.op_id()), output_num(tensor_handle.output_num()) {} in RemoteTensorHandleInternal()
28 RemoteTensorHandleInternal(int64 op_id, int32 output_num) in RemoteTensorHandleInternal()
29 : op_id(op_id), output_num(output_num) {} in RemoteTensorHandleInternal()
31 int32 output_num; member
36 return FingerprintCat64(handle.op_id, handle.output_num); in operator()
43 return first.op_id == second.op_id && first.output_num == second.output_num; in operator()
|
| D | remote_mgr.cc | 45 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetTensorHandleImpl()
68 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetMirroredResourceShape()
77 int64* op_id, int32* output_num) { in GetRemoteTensorHandle() argument
79 handle->RemoteAddress(handle->device(), op_id, output_num)); in GetRemoteTensorHandle()
82 GetTensorHandleImpl(RemoteTensorHandleInternal(*op_id, *output_num), &h)); in GetRemoteTensorHandle()
86 " and output_num:", *output_num); in GetRemoteTensorHandle()
112 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in DeleteTensorHandle()
119 int32 output_num; in SerializeRemoteTensorHandle() local
120 if (!in->RemoteAddress(device, &op_id, &output_num).ok()) { in SerializeRemoteTensorHandle()
122 TF_RETURN_IF_ERROR(GetRemoteTensorHandle(in, &op_id, &output_num)); in SerializeRemoteTensorHandle()
[all …]
|
| D | remote_mgr_test.cc | 79 const int output_num = 3; in TEST_F() local
81 op_id, output_num, t.shape(), /*remote_task=*/"", context_id_, ctx_); in TEST_F()
88 EXPECT_EQ(output_num, remote_handle.output_num()); in TEST_F()
98 const int output_num = 1; in TEST_F() local
101 op_id, output_num, t.shape(), /*remote_task=*/"", context_id_, DT_FLOAT, in TEST_F()
108 EXPECT_EQ(output_num, remote_handle.output_num()); in TEST_F()
|
| D | remote_tensor_handle_data.cc | 28 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle() argument
51 handle_to_decref->set_output_num(output_num); in DestroyRemoteTensorHandle()
86 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData() argument
92 output_num_(output_num), in RemoteTensorHandleData()
99 << ", Output num: " << output_num; in RemoteTensorHandleData()
151 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in UnshapedRemoteTensorHandleData() argument
154 output_num_(output_num), in UnshapedRemoteTensorHandleData()
161 << ", Output num: " << output_num; in UnshapedRemoteTensorHandleData()
|
| D | remote_tensor_handle_data.h | 27 RemoteTensorHandleData(int64 op_id, int output_num, const TensorShape& shape,
44 int32 output_num() const { return output_num_; } in output_num() function
60 UnshapedRemoteTensorHandleData(int64 op_id, int32 output_num,
77 int32 output_num() const { return output_num_; } in output_num() function
|
| D | eager_service_impl.h | 182 << handle_to_delete_->output_num; in Run()
192 strings::StrAppend(&out, ", output_num: ", handle_to_delete_->output_num); in DebugString()
|
| D | remote_execute_node.cc | 44 item.handle_to_decref().output_num(), ")")); in RunAsync()
|
| D | remote_mgr.h | 84 int64* op_id, int32* output_num)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 252 const xla::HloInputOutputAliasConfig& input_output_alias, int output_num) { in MustAliasOutput() argument
255 output_index = {output_num}; in MustAliasOutput()
257 DCHECK_EQ(output_num, 0) in MustAliasOutput()
258 << "output_num must be 0 for non-tuple shapes but is " << output_num; in MustAliasOutput()
271 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in FindAliasedTensorForOutput() argument
275 if (MustAliasOutput(input_output_alias, output_num)) { in FindAliasedTensorForOutput()
276 int xla_param = input_output_alias.GetAliasedParameter({output_num}) in FindAliasedTensorForOutput()
307 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in GetOrCreateTensorForOutput() argument
314 output_num, ctx, missing_ctx_input_prefix, input_output_alias, in GetOrCreateTensorForOutput()
323 const xla::HloInputOutputAliasConfig& input_output_alias, int output_num, in SetBufferForTensorUnderAllocateXlaTensors() argument
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 202 Status TensorHandle::CreateRemoteHandle(int64 op_id, int output_num, in CreateRemoteHandle() argument
209 absl::make_unique<RemoteTensorHandleData>(op_id, output_num, shape, in CreateRemoteHandle()
223 remote_output_num_(t->output_num()), in TensorHandle()
242 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in CreateUnshapedRemoteHandle() argument
245 op_id, output_num, remote_task, context_id, ctx), in CreateUnshapedRemoteHandle()
257 remote_output_num_(t->output_num()), in TensorHandle()
415 int32* output_num) const { in RemoteAddress()
421 *output_num = mirror->second->output_num(); in RemoteAddress()
428 *output_num = unshaped_mirror->second->output_num(); in RemoteAddress()
443 *output_num = remote_output_num_; in RemoteAddress()
[all …]
|
| D | tensor_handle.h | 94 static Status CreateRemoteHandle(int64 op_id, int output_num,
104 static Status CreateUnshapedRemoteHandle(int64 op_id, int32 output_num,
142 Status RemoteAddress(Device* d, int64* op_id, int32* output_num) const;
147 const int32 output_num);
|
| D | execute.cc | 707 input_handle->op_id(), input_handle->output_num(), remote_task, in EagerRemoteExecute()
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.c | 226 if (layer_configs[idx].output_num != -1) { in cnn_has_at_least_one_output()
279 const int output_num = layer_config->output_num; in av1_find_cnn_output_size() local
280 if (output_num != -1) { // Current layer is an output layer in av1_find_cnn_output_size()
281 out_width[output_num] = o_width; in av1_find_cnn_output_size()
282 out_height[output_num] = o_height; in av1_find_cnn_output_size()
283 out_channels[output_num] = channels_per_branch[layer_config->branch]; in av1_find_cnn_output_size()
876 const int output_num = layer_config->output_num; in av1_cnn_predict_c() local
877 if (output_num == -1) { // Non-output layer in av1_cnn_predict_c()
882 assign_tensor(&tensor2[branch], output[output_num], in av1_cnn_predict_c()
884 out_stride[output_num]); in av1_cnn_predict_c()
[all …]
|
| D | cnn.h | 134 int output_num; // The output buffer idx to which the layer output is member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 142 int64_t output_num = return_op.getNumOperands(); in PromoteResourcesToArguments() local
157 std::make_pair(resource_info.input_index, output_num++)); in PromoteResourcesToArguments()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 251 int64 output_num = config.fetch_size(); in PopulateXlaArgsAndXlaAlias() local
265 xla_aliases->push_back({/*output_index=*/{output_num}, in PopulateXlaArgsAndXlaAlias()
267 ++output_num; in PopulateXlaArgsAndXlaAlias()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | remote_tensor_handle.proto | 24 int32 output_num = 2; field
|
| /external/grpc-grpc/test/core/util/ |
| D | test_config.cc | 228 static void output_num(long num) { in output_num() function
242 output_num(signum); in crash_handler()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 466 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
468 last_step_tensor_outputs[i::output_num] for i in range(output_num)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 561 for (auto output_num : op.outputs) { in ConvertOp() local
562 auto& tensor = *tensors.at(output_num); in ConvertOp()
|