| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.h | 39 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()
|
| D | core.cc | 106 ImmediateExecutionTensorHandle* EagerContext::CopyTensorHandleToDevice( in CopyTensorHandleToDevice()
107 ImmediateExecutionTensorHandle* handle, const char* device_name, in CopyTensorHandleToDevice()
109 ImmediateExecutionTensorHandle* result = nullptr; in CopyTensorHandleToDevice()
155 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandle( in CreateLocalHandle()
162 ImmediateExecutionTensorHandle* EagerContext::CreateLocalHandleFromTFTensor( in CreateLocalHandleFromTFTensor()
175 ImmediateExecutionTensorHandle* EagerContext::TFTensorHandleFromInterface( in TFTensorHandleFromInterface()
176 ImmediateExecutionTensorHandle* handle) { in TFTensorHandleFromInterface()
200 for (ImmediateExecutionTensorHandle* handle : inputs_) { in Execute()
|
| D | eager_operation.h | 42 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()
201 void AddTensorHandle(ImmediateExecutionTensorHandle* h);
211 Status MaybeInferSingleInputAttrs(ImmediateExecutionTensorHandle* handle);
227 absl::InlinedVector<ImmediateExecutionTensorHandle*, 4> inputs_;
265 ImmediateExecutionTensorHandle** slot = &inputs_[i]; in UpdateInput()
266 ImmediateExecutionTensorHandle* existing = *slot; in UpdateInput()
|
| D | custom_device_test.cc | 35 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice()
36 ImmediateExecutionTensorHandle** result) override { in CopyTensorToDevice()
42 ImmediateExecutionTensorHandle* tensor, in CopyTensorFromDevice()
44 ImmediateExecutionTensorHandle** result) override { in CopyTensorFromDevice()
50 ImmediateExecutionTensorHandle** retvals, in Execute()
55 Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles, in Pack()
56 ImmediateExecutionTensorHandle** result) override { in Pack()
|
| D | custom_device_op_handler.cc | 55 ImmediateExecutionTensorHandle** retvals, in Execute()
81 tensorflow::ImmediateExecutionTensorHandle* new_tesnor; in Execute()
115 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
138 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
|
| D | eager_operation.cc | 35 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()
353 ImmediateExecutionTensorHandle* handle) { in MaybeInferSingleInputAttrs()
509 void EagerOperation::AddTensorHandle(ImmediateExecutionTensorHandle* h) { in AddTensorHandle()
|
| D | context.h | 125 ImmediateExecutionTensorHandle* CreateLocalHandle(
128 ImmediateExecutionTensorHandle* CreateLocalHandleFromTFTensor(
130 ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
131 ImmediateExecutionTensorHandle* handle, const char* device_name,
139 ImmediateExecutionTensorHandle* TFTensorHandleFromInterface(
140 ImmediateExecutionTensorHandle* handle) override;
|
| D | placement_utils.h | 39 absl::Span<ImmediateExecutionTensorHandle* const> args,
|
| D | tensor_handle.h | 56 class TensorHandle : public ImmediateExecutionTensorHandle {
136 ImmediateExecutionTensorHandle* Copy() override;
394 class TensorHandleInterface : public ImmediateExecutionTensorHandle {
|
| D | custom_device_op_handler.h | 37 ImmediateExecutionTensorHandle** retvals, int* num_retvals);
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.cc | 62 if (!tensorflow::isa<ImmediateExecutionTensorHandle>( in CreateUninitializedResourceVariable()
66 handle->reset(reinterpret_cast<ImmediateExecutionTensorHandle*>( in CreateUninitializedResourceVariable()
72 ImmediateExecutionTensorHandle* variable_handle, in AssignVariable()
73 DataType dtype, ImmediateExecutionTensorHandle* value) { in AssignVariable()
86 ImmediateExecutionTensorHandle* variable_handle, in ReadVariable()
98 if (!tensorflow::isa<ImmediateExecutionTensorHandle>(owned_value.get())) { in ReadVariable()
102 reinterpret_cast<ImmediateExecutionTensorHandle*>(owned_value.release())); in ReadVariable()
107 ImmediateExecutionTensorHandle* handle) { in DestroyResource()
|
| D | variable_ops.h | 43 ImmediateExecutionTensorHandle* variable_handle,
44 DataType dtype, ImmediateExecutionTensorHandle* value);
50 ImmediateExecutionTensorHandle* variable_handle,
57 ImmediateExecutionTensorHandle* handle);
|
| D | restore_ops.cc | 101 if (!tensorflow::isa<ImmediateExecutionTensorHandle>( in SingleRestore()
105 out->reset(reinterpret_cast<ImmediateExecutionTensorHandle*>( in SingleRestore()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_tensor_handle.h | 34 class ImmediateExecutionTensorHandle : public AbstractTensorHandle {
55 virtual ImmediateExecutionTensorHandle* Copy() = 0;
71 explicit ImmediateExecutionTensorHandle(AbstractTensorHandleKind kind) in ImmediateExecutionTensorHandle() function
73 ~ImmediateExecutionTensorHandle() override {} in ~ImmediateExecutionTensorHandle()
78 void operator()(ImmediateExecutionTensorHandle* p) const { in operator()
87 std::unique_ptr<ImmediateExecutionTensorHandle,
|
| D | immediate_execution_context.h | 91 virtual ImmediateExecutionTensorHandle* CreateLocalHandle(
94 virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
95 ImmediateExecutionTensorHandle* handle, const char* device_name,
198 virtual ImmediateExecutionTensorHandle* CreateLocalHandleFromTFTensor(
202 virtual ImmediateExecutionTensorHandle* TFTensorHandleFromInterface(
203 ImmediateExecutionTensorHandle* handle) = 0;
|
| D | tfe_tensorhandle_internal.h | 31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle,
33 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionTensorHandle*,
|
| D | c_api_unified_experimental_eager.cc | 40 using tensorflow::ImmediateExecutionTensorHandle;
62 auto handle = dyn_cast<ImmediateExecutionTensorHandle>(unwrap(at)); in TF_AbstractTensorGetEagerTensor()
|
| D | immediate_execution_operation.h | 45 virtual absl::Span<ImmediateExecutionTensorHandle* const> GetInputs()
48 ImmediateExecutionTensorHandle* input) = 0;
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | partially_revived_objects.cc | 82 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()
|
| D | flat_tensor_function.cc | 50 std::vector<ImmediateExecutionTensorHandle*> captures, in Create()
55 for (ImmediateExecutionTensorHandle* capture : captures) { in Create()
|
| D | tf_concrete_function_revival_state.h | 43 std::vector<ImmediateExecutionTensorHandle*> captures;
|
| D | tf_signature_def_function_revival_state.h | 50 std::vector<ImmediateExecutionTensorHandle*> captures;
|
| D | constant.cc | 38 ImmediateExecutionTensorHandle* handle = ctx->CreateLocalHandle(tensor); in Create()
|
| D | tensorhandle_convertible.h | 33 ImmediateExecutionTensorHandle* handle() { return handle_.get(); } in handle()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 60 using tensorflow::ImmediateExecutionTensorHandle;
231 if (!isa<ImmediateExecutionTensorHandle>(self)) { in PYBIND11_MODULE()
240 wrap(dyn_cast<ImmediateExecutionTensorHandle>(self)); in PYBIND11_MODULE()
|