| /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 | eager_service_impl.h | 140 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetTensorHandle()
154 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in DeleteTensorHandle()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 299 int output_num = 0; in PopulateOutputs() local
354 se::DeviceMemoryBase buffer = output.buffer({output_num}); in PopulateOutputs()
360 xla_tensor->set_shaped_buffer(output.TakeSubTree({output_num})); in PopulateOutputs()
372 output.set_buffer(xla::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
375 ++output_num; in PopulateOutputs()
425 xla_tensor->set_shaped_buffer(output.TakeSubTree({output_num})); in PopulateOutputs()
432 se::DeviceMemoryBase buffer = output.buffer({output_num}); in PopulateOutputs()
433 output.set_buffer(xla::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
438 ++output_num; in PopulateOutputs()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.h | 573 inline const float& left_sum(int split, int output_num) const { in left_sum() argument
574 return left_sums_[split * num_outputs_ + output_num]; in left_sum()
576 inline float& left_sum(int split, int output_num) { in left_sum() argument
577 return left_sums_[split * num_outputs_ + output_num]; in left_sum()
579 inline const float& left_square(int split, int output_num) const { in left_square() argument
580 return left_squares_[split * num_outputs_ + output_num]; in left_square()
582 inline float& left_square(int split, int output_num) { in left_square() argument
583 return left_squares_[split * num_outputs_ + output_num]; in left_square()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 81 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle() argument
92 remote_output_num_(output_num), in TensorHandle()
98 << ", Output num: " << output_num; in TensorHandle()
232 Status TensorHandle::RemoteAddress(int64* op_id, int32* output_num) { in RemoteAddress() argument
238 *output_num = remote_output_num_; in RemoteAddress()
|
| D | tensor_handle.h | 66 TensorHandle(int64 op_id, int32 output_num, uint64 remote_shape_node_id,
99 Status RemoteAddress(int64* op_id, int32* output_num);
|
| D | execute.cc | 530 uint64 op_id, int output_num) { in GetRemoteTensorDestructor() argument
531 return [ctx, eager_client, context_id, op_id, output_num]() { in GetRemoteTensorDestructor()
544 handle_to_decref->set_output_num(output_num); in GetRemoteTensorDestructor()
683 int32 output_num; in EagerRemoteExecute()
684 TF_RETURN_IF_ERROR(input->RemoteAddress(&op_id, &output_num)); in EagerRemoteExecute()
688 remote_op_input->set_output_num(output_num); in EagerRemoteExecute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 270 int64 output_num = config.fetch_size(); in ConvertGraphToXla() local
283 xla_aliases.push_back({/*output_index=*/{output_num}, in ConvertGraphToXla()
286 ++output_num; in ConvertGraphToXla()
|
| /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 | 372 output_num = len(last_step_tensor_outputs) // self._num_replicas_in_sync
374 last_step_tensor_outputs[i::output_num] for i in range(output_num)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | eager_service.proto | 17 int32 output_num = 2; field
|