| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | core.cc | 132 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle() function in tensorflow::EagerContext
135 return TensorHandle::CreateLocalHandle(std::move(tensor), /*d=*/HostCPU(), in CreateLocalHandle()
143 return TensorHandle::CreateLocalHandle(std::move(t), /*d=*/HostCPU(), in CreateLocalHandleFromTFTensor()
148 return TensorHandle::CreateLocalHandle(std::move(t), /*d=*/d, in CreateLocalHandleFromTFTensor()
|
| D | tensor_handle_test.cc | 44 TensorHandle::CreateLocalHandle(std::move(t), nullptr, nullptr, ctx); in TEST()
149 TensorHandle::CreateLocalHandle(std::move(t0), d0, d0, d0, context()); in TEST_F()
155 TensorHandle::CreateLocalHandle(std::move(t1), d1, d1, d1, context()); in TEST_F()
225 TensorHandle::CreateLocalHandle(std::move(t), d, d, d, context()); in TEST_F()
266 TensorHandle::CreateLocalHandle(std::move(t), d, d, d, ctx); in TEST()
429 TensorHandle::CreateLocalHandle(std::move(tcpu), dcpu, dcpu, dcpu, ctx); in TEST()
445 TensorHandle::CreateLocalHandle(std::move(tgpu), dgpu, dgpu, dgpu, ctx); in TEST()
|
| D | execute_node_test.cc | 94 TensorHandle::CreateLocalHandle(std::move(t0), device0, device0, ctx); in TEST()
98 TensorHandle::CreateLocalHandle(std::move(t1), device0, device0, ctx); in TEST()
|
| D | tensor_handle.h | 76 static TensorHandle* CreateLocalHandle(const tensorflow::Tensor& t);
77 static TensorHandle* CreateLocalHandle(tensorflow::Tensor&& t, Device* d,
79 static TensorHandle* CreateLocalHandle(tensorflow::Tensor&& t, Device* d,
|
| D | custom_device_test.cc | 164 TensorHandle::CreateLocalHandle(std::move(resource_tensor), in TEST()
169 TensorHandle::CreateLocalHandle(std::move(float_tensor), physical_device, in TEST()
|
| D | tensor_handle.cc | 216 TensorHandle* TensorHandle::CreateLocalHandle(const tensorflow::Tensor& t) { in CreateLocalHandle() function in tensorflow::TensorHandle
219 return CreateLocalHandle(std::move(tensor), in CreateLocalHandle()
225 TensorHandle* TensorHandle::CreateLocalHandle(tensorflow::Tensor&& t, Device* d, in CreateLocalHandle() function in tensorflow::TensorHandle
228 return CreateLocalHandle(std::move(t), d, op_device, nullptr, ctx); in CreateLocalHandle()
231 TensorHandle* TensorHandle::CreateLocalHandle(tensorflow::Tensor&& t, Device* d, in CreateLocalHandle() function in tensorflow::TensorHandle
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | restore_ops.cc | 44 out->reset(ctx->CreateLocalHandle(tensor.get())); in CreateStringScalarTensorHandle()
63 out->reset(ctx->CreateLocalHandle(tensor.get())); in CreateStringVectorTensorHandle()
|
| D | variable_ops_test.cc | 36 ImmediateTensorHandlePtr handle(context->CreateLocalHandle(tensor.get())); in CreateScalarTensorHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_tensor.cc | 180 tensorflow::TensorHandle::CreateLocalHandle(tensor)}; in MoveDHTToRuntimeFallbackTensor()
194 tensorflow::TensorHandle::CreateLocalHandle(tensor)}; in CopyRefDHTToRuntimeFallbackTensor()
204 tensorflow::TensorHandle::CreateLocalHandle(tensor)}; in CopySHTToRuntimeFallbackTensor()
|
| D | runtime_fallback_kernels.cc | 1132 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in RuntimeFallbackExecuteOp()
1207 tensorflow::TensorHandle::CreateLocalHandle(tensor)}; in ConvertTFRTTensorToTFTensorHandle()
1213 tensorflow::TensorHandle::CreateLocalHandle(tensor)}; in ConvertTFRTTensorToTFTensorHandle()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | constant.cc | 38 ImmediateExecutionTensorHandle* handle = ctx->CreateLocalHandle(tensor); in Create()
|
| D | asset.cc | 44 ImmediateTensorHandlePtr handle(ctx->CreateLocalHandle(tensor.get())); in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 75 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
114 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
|
| D | eager_service_impl.cc | 129 handle = TensorHandle::CreateLocalHandle(std::move(tensor), nullptr, in GetEagerOperationAndNumRetvals()
723 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor()
761 handles[i] = TensorHandle::CreateLocalHandle( in SendPackedHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| D | conversion.cc | 51 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in ConvertKernelFallbackToRuntimeFallbackTensor()
|
| /external/tensorflow/tensorflow/cc/experimental/libtf/runtime/ |
| D | runtime.h | 100 impl::TaggedValueTensor(ctx->CreateLocalHandle(t.get()), false)))); in CreateHostTensor()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad.cc | 82 ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); in BroadcastMul()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 85 ImmediateTensorHandlePtr handle(ctx->CreateLocalHandle(tensor.get())); in CreateTensorHandle()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 92 virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
|
| D | c_api.cc | 276 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); in TFE_NewTensorHandle()
588 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle( in TFE_NewTensorHandleFromDeviceMemory()
951 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(t)); in TFE_NewTensorHandle()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_seq_tensor.cc | 309 *h = tensorflow::wrap(tensorflow::unwrap(ctx)->CreateLocalHandle(t)); in Convert()
702 tensorflow::unwrap(ctx)->CreateLocalHandle(tf_tensor->tensor)); in NumpyToTFE_TensorHandle()
888 tensorflow::wrap(tensorflow::unwrap(ctx)->CreateLocalHandle(t)); in PySeqToTFE_TensorHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 55 OwnedTensorHandle tensor_handle{tensorflow::TensorHandle::CreateLocalHandle( in CopyRefGpuTensorToRuntimeFallbackTensor()
|
| /external/tensorflow/tensorflow/core/function/ |
| D | runtime_client_test.cc | 81 ImmediateTensorHandlePtr handle(ctx.CreateLocalHandle(tensor.get())); in IntScalarTensor()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 771 tensorflow::ImmediateExecutionTensorHandle* ContextInterface::CreateLocalHandle( in CreateLocalHandle() function in tfrt::tf::ContextInterface
780 tensorflow::TensorHandle::CreateLocalHandle( in CreateLocalHandle()
856 tensorflow::TensorHandle::CreateLocalHandle(std::move(t))}; in CreateLocalHandleFromTFTensor()
892 return tensorflow::TensorHandle::CreateLocalHandle( in TFTensorHandleFromInterface()
898 return tensorflow::TensorHandle::CreateLocalHandle( in TFTensorHandleFromInterface()
|
| D | c_api_tfrt.h | 101 tensorflow::ImmediateExecutionTensorHandle* CreateLocalHandle(
|