Home
last modified time | relevance | path

Searched defs:send_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc1079 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local
1152 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local
1257 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local
H A Dremote_copy_node.cc431 auto* send_tensor = request.add_queue()->mutable_send_tensor(); in StartRemoteSendTensor() local
H A Deager_service_impl.cc714 Status EagerServiceImpl::SendTensor(const SendTensorOp& send_tensor, in SendTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Deager_service.proto64 SendTensorOp send_tensor = 3; field
/aosp_15_r20/external/pytorch/test/
H A Dtest_multiprocessing.py88 def send_tensor(queue, event, device, dtype): function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Deager_service.pb.cc1261 QueueItem::_Internal::send_tensor(const QueueItem* msg) { in send_tensor() function in tensorflow::eager::QueueItem::_Internal
1319 void QueueItem::set_allocated_send_tensor(::tensorflow::eager::SendTensorOp* send_tensor) { in set_allocated_send_tensor()
Deager_service.pb.h5022 inline const ::tensorflow::eager::SendTensorOp& QueueItem::send_tensor() const { in send_tensor() function
5037 …id QueueItem::unsafe_arena_set_allocated_send_tensor(::tensorflow::eager::SendTensorOp* send_tenso… in unsafe_arena_set_allocated_send_tensor()