| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | core.cc | 155 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle() function in tensorflow::EagerContext
158 return TensorHandle::CreateLocalHandle(std::move(tensor), /*d=*/HostCPU(), in CreateLocalHandle()
166 return TensorHandle::CreateLocalHandle(std::move(t), /*d=*/HostCPU(), in CreateLocalHandleFromTFTensor()
171 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 | 89 TensorHandle::CreateLocalHandle(std::move(t0), device0, device0, ctx); in TEST()
93 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 | 129 TensorHandle::CreateLocalHandle(std::move(resource_tensor), in TEST()
134 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
|
| D | context.h | 125 ImmediateExecutionTensorHandle* CreateLocalHandle(
|
| D | execute.cc | 996 retvals[i] = TensorHandle::CreateLocalHandle( in GetKernelOutputs()
|
| /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/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 72 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
111 TensorHandle* handle = TensorHandle::CreateLocalHandle( in TEST_F()
|
| D | eager_service_impl.cc | 129 handle = TensorHandle::CreateLocalHandle(std::move(tensor), nullptr, in GetEagerOperationAndNumRetvals()
684 TensorHandle* tensor_handle = TensorHandle::CreateLocalHandle( in SendTensor()
722 handles[i] = TensorHandle::CreateLocalHandle( in SendPackedHandle()
|
| /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/c/experimental/saved_model/core/ |
| D | test_utils.cc | 85 ImmediateTensorHandlePtr handle(ctx->CreateLocalHandle(tensor.get())); in CreateTensorHandle()
|
| /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/eager/ |
| D | immediate_execution_context.h | 91 virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
|
| D | c_api.cc | 271 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); in TFE_NewTensorHandle()
607 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle( in TFE_NewTensorHandleFromDeviceMemory()
965 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(t)); in TFE_NewTensorHandle()
|
| D | c_api_experimental.cc | 605 tensorflow::unwrap(ctx)->CreateLocalHandle(t->tensor)); in TFE_NewTensorHandleFromTensor()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_seq_tensor.cc | 309 *h = tensorflow::wrap(tensorflow::unwrap(ctx)->CreateLocalHandle(t)); in Convert()
701 tensorflow::unwrap(ctx)->CreateLocalHandle(tf_tensor->tensor)); in NumpyToTFE_TensorHandle()
887 tensorflow::wrap(tensorflow::unwrap(ctx)->CreateLocalHandle(t)); in PySeqToTFE_TensorHandle()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 288 tensorflow::TensorHandle::CreateLocalHandle( in BuildEagerInputs()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 496 return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); in TFE_NewTensorHandleFromScalar()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 1098 tensorflow::wrap(tensorflow::unwrap(ctx)->CreateLocalHandle(tensor)); in TFE_Py_TensorShapeSlice()
|