| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | conversion_function.cc | 46 tensor.GetTensorHandle()->Resolve(&status)}; in ConvertRuntimeFallbackTensorToDenseHostTensor()
72 tensor.GetTensorHandle()->Resolve(&status)}; in ConvertRuntimeFallbackTensorToStringHostTensor()
150 auto *th = tensor.GetTensorHandle(); in TransferRuntimeFallbackToAnotherDevice()
|
| D | runtime_fallback_tensor.h | 42 TensorHandle* GetTensorHandle() const { return tensor_handle_.get(); } in GetTensorHandle() function
|
| D | runtime_fallback_op_handler.cc | 132 result_tensor.GetTensorHandle()->DeviceName(&status); in GetDeviceFromFallbackTensor()
270 result_tensor_av->get<RuntimeFallbackTensor>().GetTensorHandle(); in MakeOp()
|
| D | runtime_fallback_kernels.cc | 603 llvm::cast<RuntimeFallbackTensor>(input_tensor)->GetTensorHandle()); in RuntimeFallbackExecute()
713 input_tensor_av->get<RuntimeFallbackTensor>().GetTensorHandle(); in RuntimeFallbackKernel()
1168 runtime_fallback_tensor.GetTensorHandle()->Tensor(&tf_tensor); in RuntimeFallbackExecuteOp()
|
| D | runtime_fallback_batch_tf_opkernels.cc | 74 Status s = rtfbt->GetTensorHandle()->Tensor(&tf_tensor); in ConvertTFRTTensorToTFTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 62 Status GetTensorHandle(const uint64 context_id, in GetTensorHandle() function in tensorflow::eager::__anon7a15dd630111::TestEagerServiceImpl
69 return context->Context()->RemoteMgr()->GetTensorHandle(remote_handle, in GetTensorHandle()
446 TF_ASSERT_OK(eager_service_impl.GetTensorHandle( in TEST_F()
554 TF_ASSERT_OK(eager_service_impl.GetTensorHandle( in TestFunction()
647 TF_ASSERT_OK(eager_service_impl.GetTensorHandle( in TestComponentFunction()
818 TF_ASSERT_OK(eager_service_impl_.GetTensorHandle( in CheckOutputsAndClose()
925 TF_ASSERT_OK(eager_service_impl_.GetTensorHandle( in TEST_F()
1102 TF_ASSERT_OK(eager_service_impl.GetTensorHandle( in TEST_F()
1182 TF_ASSERT_OK(eager_service_impl.GetTensorHandle( in TEST_F()
|
| D | remote_mgr.cc | 82 Status RemoteMgr::GetTensorHandle( in GetTensorHandle() function in tensorflow::eager::RemoteMgr
187 TF_RETURN_IF_ERROR(GetTensorHandle(RemoteTensorHandleInternal(in), out)); in DeserializeRemoteTensorHandle()
|
| D | remote_mgr.h | 52 Status GetTensorHandle(const RemoteTensorHandleInternal& remote_handle,
|
| D | remote_mgr_test.cc | 178 s = remote_mgr.GetTensorHandle(remote_handle_internal, &out); in TEST_F()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| D | conversion.cc | 67 Status s = tensor.GetTensorHandle()->Tensor(&tf_tensor); in ConvertRuntimeFallbackToKernelFallbackTensor()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 72 auto tf_tensor_handle = tensor.GetTensorHandle(); in ConvertRuntimeFallbackTensorToDenseGpuTensor()
|
| /external/armnn/src/armnn/ |
| D | LoadedNetwork.cpp | 759 ITensorHandle* GetTensorHandle() const { return m_TensorHandle.get(); } in GetTensorHandle() function in armnn::__anon8d9f67980511::TensorPin
908 EnqueueInput(*inputLayer, pin.GetTensorHandle(), pin.GetTensorInfo()); in EnqueueWorkload()
988 EnqueueOutput(*outputLayer, pin.GetTensorHandle(), pin.GetTensorInfo()); in EnqueueWorkload()
1961 auto GetTensorHandle = [&](Layer* layer, const OutputSlot& outputSlot) in CreateWorkingMemHandle() local
2052 managedTensorHandles.emplace_back(GetTensorHandle(layer, slot)); in CreateWorkingMemHandle()
2057 unmanagedTensorHandles.emplace_back(GetTensorHandle(layer, slot)); in CreateWorkingMemHandle()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 300 .GetTensorHandle() in Type()
396 .GetTensorHandle() in DataType()
889 return rtfbt->GetTensorHandle(); in TFTensorHandleFromInterface()
|