Home
last modified time | relevance | path

Searched refs:ImmediateExecutionTensorHandle (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device.h39 ImmediateExecutionTensorHandle* tensor,
40 ImmediateExecutionTensorHandle** result) = 0;
43 ImmediateExecutionTensorHandle* tensor, const string& target_device_name,
44 ImmediateExecutionTensorHandle** result) = 0;
47 ImmediateExecutionTensorHandle** retvals,
52 virtual Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles,
53 ImmediateExecutionTensorHandle** result) = 0;
72 class CustomDeviceTensorHandle : public ImmediateExecutionTensorHandle {
76 : ImmediateExecutionTensorHandle(kCustomDevice), in CustomDeviceTensorHandle()
101 ImmediateExecutionTensorHandle* Copy() override { in Copy()
Dcore.cc106 ImmediateExecutionTensorHandle* EagerContext::CopyTensorHandleToDevice( in CopyTensorHandleToDevice()
107 ImmediateExecutionTensorHandle* handle, const char* device_name, in CopyTensorHandleToDevice()
109 ImmediateExecutionTensorHandle* result = nullptr; in CopyTensorHandleToDevice()
132 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle()
139 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandleFromTFTensor( in CreateLocalHandleFromTFTensor()
152 ImmediateExecutionTensorHandle* EagerContext::TFTensorHandleFromInterface( in TFTensorHandleFromInterface()
153 ImmediateExecutionTensorHandle* handle) { in TFTensorHandleFromInterface()
177 for (ImmediateExecutionTensorHandle* handle : inputs_) { in Execute()
Dcustom_device_op_handler.cc55 ImmediateExecutionTensorHandle** retvals, in Execute()
81 tensorflow::ImmediateExecutionTensorHandle* new_tensor; in Execute()
98 ImmediateExecutionTensorHandle* CustomDeviceOpHandler::CopyTensorHandleToDevice( in CopyTensorHandleToDevice()
99 ImmediateExecutionContext* context, ImmediateExecutionTensorHandle* handle, in CopyTensorHandleToDevice()
102 ImmediateExecutionTensorHandle* result = nullptr; in CopyTensorHandleToDevice()
148 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
171 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
Deager_operation.h42 for (ImmediateExecutionTensorHandle* h : inputs_) { in ~EagerOperation()
88 Status SetInput(size_t index, ImmediateExecutionTensorHandle* input) override;
89 absl::Span<ImmediateExecutionTensorHandle* const> GetInputs() const override;
162 const absl::InlinedVector<ImmediateExecutionTensorHandle*, 4>& Inputs() in Inputs()
225 void AddTensorHandle(ImmediateExecutionTensorHandle* h);
235 Status MaybeInferSingleInputAttrs(ImmediateExecutionTensorHandle* handle);
251 absl::InlinedVector<ImmediateExecutionTensorHandle*, 4> inputs_;
290 ImmediateExecutionTensorHandle** slot = &inputs_[i]; in UpdateInput()
291 ImmediateExecutionTensorHandle* existing = *slot; in UpdateInput()
Dcustom_device_test.cc38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice()
39 ImmediateExecutionTensorHandle** result) override { in CopyTensorToDevice()
45 ImmediateExecutionTensorHandle* tensor, in CopyTensorFromDevice()
47 ImmediateExecutionTensorHandle** result) override { in CopyTensorFromDevice()
53 ImmediateExecutionTensorHandle** retvals, in Execute()
58 Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles, in Pack()
59 ImmediateExecutionTensorHandle** result) override { in Pack()
Dcustom_device_op_handler.h37 ImmediateExecutionTensorHandle** retvals, int* num_retvals);
39 ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
41 ImmediateExecutionTensorHandle* handle, const char* device_name,
Deager_operation.cc35 for (ImmediateExecutionTensorHandle* h : inputs_) { in Clear()
240 absl::Span<ImmediateExecutionTensorHandle* const> EagerOperation::GetInputs() in GetInputs()
244 reinterpret_cast<ImmediateExecutionTensorHandle* const*>(inputs_.data()), in GetInputs()
268 ImmediateExecutionTensorHandle* h = in AddInput()
269 down_cast<ImmediateExecutionTensorHandle*>(input); in AddInput()
286 ImmediateExecutionTensorHandle* h = in AddInputList()
287 down_cast<ImmediateExecutionTensorHandle*>(input); in AddInputList()
294 ImmediateExecutionTensorHandle* input) { in SetInput()
355 ImmediateExecutionTensorHandle* handle) { in MaybeInferSingleInputAttrs()
511 void EagerOperation::AddTensorHandle(ImmediateExecutionTensorHandle* h) { in AddTensorHandle()
Dcontext_test.cc270 auto input_float = core::RefCountPtr<ImmediateExecutionTensorHandle>( in TEST_F()
274 auto input_bool_false = core::RefCountPtr<ImmediateExecutionTensorHandle>( in TEST_F()
293 auto input_bool_true = core::RefCountPtr<ImmediateExecutionTensorHandle>( in TEST_F()
330 auto input_int = core::RefCountPtr<ImmediateExecutionTensorHandle>( in TEST_F()
Dtensor_handle.h56 class TensorHandle : public ImmediateExecutionTensorHandle {
136 ImmediateExecutionTensorHandle* Copy() override;
398 class TensorHandleInterface : public ImmediateExecutionTensorHandle {
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc61 if (!tensorflow::isa<ImmediateExecutionTensorHandle>( in CreateUninitializedResourceVariable()
65 handle->reset(reinterpret_cast<ImmediateExecutionTensorHandle*>( in CreateUninitializedResourceVariable()
71 ImmediateExecutionTensorHandle* variable_handle, in AssignVariable()
72 DataType dtype, ImmediateExecutionTensorHandle* value) { in AssignVariable()
85 ImmediateExecutionTensorHandle* variable_handle, in ReadVariable()
97 if (!tensorflow::isa<ImmediateExecutionTensorHandle>(owned_value.get())) { in ReadVariable()
101 reinterpret_cast<ImmediateExecutionTensorHandle*>(owned_value.release())); in ReadVariable()
106 ImmediateExecutionTensorHandle* handle) { in DestroyResource()
Dvariable_ops.h43 ImmediateExecutionTensorHandle* variable_handle,
44 DataType dtype, ImmediateExecutionTensorHandle* value);
50 ImmediateExecutionTensorHandle* variable_handle,
57 ImmediateExecutionTensorHandle* handle);
Drestore_ops.cc101 if (!tensorflow::isa<ImmediateExecutionTensorHandle>( in SingleRestore()
105 out->reset(reinterpret_cast<ImmediateExecutionTensorHandle*>( in SingleRestore()
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_tensor_handle.h34 class ImmediateExecutionTensorHandle : public AbstractTensorHandle {
60 virtual ImmediateExecutionTensorHandle* Copy() = 0;
95 explicit ImmediateExecutionTensorHandle(AbstractTensorHandleKind kind) in ImmediateExecutionTensorHandle() function
97 ~ImmediateExecutionTensorHandle() override {} in ~ImmediateExecutionTensorHandle()
102 void operator()(ImmediateExecutionTensorHandle* p) const { in operator()
111 std::unique_ptr<ImmediateExecutionTensorHandle,
Dimmediate_execution_context.h92 virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
95 virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
96 ImmediateExecutionTensorHandle* handle, const char* device_name,
227 virtual ImmediateExecutionTensorHandle* CreateLocalHandleFromTFTensor(
231 virtual ImmediateExecutionTensorHandle* TFTensorHandleFromInterface(
232 ImmediateExecutionTensorHandle* handle) = 0;
Dimmediate_execution_tensor_handle.cc20 std::string ImmediateExecutionTensorHandle::DebugString() const { in DebugString()
47 Status ImmediateExecutionTensorHandle::SummarizeValue( in SummarizeValue()
53 const_cast<ImmediateExecutionTensorHandle*>(this)->Resolve(&status)); in SummarizeValue()
Dtfe_tensorhandle_internal.h31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle,
33 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle*,
Dc_api_unified_experimental_eager.cc40 using tensorflow::ImmediateExecutionTensorHandle;
62 auto handle = dyn_cast<ImmediateExecutionTensorHandle>(unwrap(at)); in TF_AbstractTensorGetEagerTensor()
Dimmediate_execution_operation.h46 virtual absl::Span<ImmediateExecutionTensorHandle* const> GetInputs()
49 ImmediateExecutionTensorHandle* input) = 0;
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h101 tensorflow::ImmediateExecutionTensorHandle* CreateLocalHandle(
104 tensorflow::ImmediateExecutionTensorHandle* CreateLocalHandleFromTFTensor(
108 tensorflow::ImmediateExecutionTensorHandle* TFTensorHandleFromInterface(
109 tensorflow::ImmediateExecutionTensorHandle* handle) override;
111 tensorflow::ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
112 tensorflow::ImmediateExecutionTensorHandle* handle,
354 : public tensorflow::ImmediateExecutionTensorHandle {
392 tensorflow::ImmediateExecutionTensorHandle* Copy() override { in Copy()
506 size_t index, tensorflow::ImmediateExecutionTensorHandle* input) override;
507 absl::Span<tensorflow::ImmediateExecutionTensorHandle* const> GetInputs()
[all …]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dpartially_revived_objects.cc82 ImmediateExecutionTensorHandle** handle) { in TensorHandleFromNode()
285 std::vector<ImmediateExecutionTensorHandle*> captures; in CreateConcreteFunction()
288 ImmediateExecutionTensorHandle* capture_handle; in CreateConcreteFunction()
307 std::vector<ImmediateExecutionTensorHandle*> captures; in CreateSignatureDefFunction()
310 ImmediateExecutionTensorHandle* capture_handle; in CreateSignatureDefFunction()
417 if (!tensorflow::isa<ImmediateExecutionTensorHandle>( in CreateAllResourceHandles()
422 reinterpret_cast<ImmediateExecutionTensorHandle*>( in CreateAllResourceHandles()
Dflat_tensor_function.cc50 std::vector<ImmediateExecutionTensorHandle*> captures, in Create()
55 for (ImmediateExecutionTensorHandle* capture : captures) { in Create()
Dtf_concrete_function_revival_state.h43 std::vector<ImmediateExecutionTensorHandle*> captures;
Dtf_signature_def_function_revival_state.h50 std::vector<ImmediateExecutionTensorHandle*> captures;
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc107 class FakeTensorHandle : public tensorflow::ImmediateExecutionTensorHandle {
110 : ImmediateExecutionTensorHandle(kTfrt), in FakeTensorHandle()
149 ImmediateExecutionTensorHandle* Copy() override { in Copy()
198 tensorflow::ImmediateExecutionTensorHandle* input) override { in SetInput()
204 absl::Span<tensorflow::ImmediateExecutionTensorHandle* const> GetInputs() in GetInputs()
207 reinterpret_cast<tensorflow::ImmediateExecutionTensorHandle* const*>( in GetInputs()
/external/tensorflow/tensorflow/python/framework/experimental/
Dunified_api.cc61 using tensorflow::ImmediateExecutionTensorHandle;
233 if (!isa<ImmediateExecutionTensorHandle>(self)) { in PYBIND11_MODULE()
242 wrap(dyn_cast<ImmediateExecutionTensorHandle>(self)); in PYBIND11_MODULE()

123