Searched refs:ImmediateTensorHandlePtr (Results 1 – 25 of 25) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops_test.cc | 33 ImmediateTensorHandlePtr CreateScalarTensorHandle(EagerContext* context, in CreateScalarTensorHandle()
36 ImmediateTensorHandlePtr handle(context->CreateLocalHandle(tensor.get())); in CreateScalarTensorHandle()
56 ImmediateTensorHandlePtr handle; in TEST_F()
66 ImmediateTensorHandlePtr handle; in TEST_F()
77 ImmediateTensorHandlePtr variable; in TEST_F()
83 ImmediateTensorHandlePtr my_value = CreateScalarTensorHandle(context(), 42.0); in TEST_F()
88 ImmediateTensorHandlePtr read_value_handle; in TEST_F()
|
| D | restore_ops.cc | 37 ImmediateTensorHandlePtr* out) { in CreateStringScalarTensorHandle()
51 ImmediateTensorHandlePtr* out) { in CreateStringVectorTensorHandle()
71 ImmediateTensorHandlePtr* out) { in SingleRestore()
77 ImmediateTensorHandlePtr prefix_handle; in SingleRestore()
81 ImmediateTensorHandlePtr names_handle; in SingleRestore()
88 ImmediateTensorHandlePtr shapes_and_slices_handle; in SingleRestore()
|
| D | restore_ops_test.cc | 63 ImmediateTensorHandlePtr x_handle; in TEST_F()
73 ImmediateTensorHandlePtr y_handle; in TEST_F()
83 ImmediateTensorHandlePtr z_handle; in TEST_F()
95 ImmediateTensorHandlePtr x_handle; in TEST_F()
103 ImmediateTensorHandlePtr x_handle; in TEST_F()
|
| D | variable_ops.h | 35 ImmediateTensorHandlePtr* handle);
51 DataType dtype, ImmediateTensorHandlePtr* output);
|
| D | variable_ops.cc | 41 ImmediateTensorHandlePtr* handle) { in CreateUninitializedResourceVariable()
87 DataType dtype, ImmediateTensorHandlePtr* output) { in ReadVariable()
|
| D | restore_ops.h | 35 ImmediateTensorHandlePtr* out);
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.cc | 34 ImmediateTensorHandlePtr handle) in Variable()
63 Status Variable::ReadValue(ImmediateTensorHandlePtr* out) { in ReadValue()
72 ImmediateTensorHandlePtr handle; in CreateUninitialized()
91 ImmediateTensorHandlePtr handlePtr; in CreateUninitialized()
|
| D | flat_tensor_function.cc | 36 const std::string& name, std::vector<ImmediateTensorHandlePtr> captures, in FlatTensorFunction()
53 std::vector<ImmediateTensorHandlePtr> owned_captures; in Create()
57 owned_captures.push_back(ImmediateTensorHandlePtr(capture)); in Create()
|
| D | constant.cc | 32 Constant::Constant(ImmediateTensorHandlePtr handle) in Constant()
42 output->reset(new Constant(ImmediateTensorHandlePtr(handle))); in Create()
|
| D | tensorhandle_convertible.h | 30 explicit TensorHandleConvertible(ImmediateTensorHandlePtr handle) in TensorHandleConvertible()
44 ImmediateTensorHandlePtr handle_;
|
| D | flat_tensor_function.h | 72 std::vector<ImmediateTensorHandlePtr> captures,
80 std::vector<ImmediateTensorHandlePtr> captures_;
|
| D | asset.cc | 29 Asset::Asset(ImmediateTensorHandlePtr handle) in Asset()
44 ImmediateTensorHandlePtr handle(ctx->CreateLocalHandle(tensor.get())); in Create()
|
| D | variable.h | 53 Status ReadValue(ImmediateTensorHandlePtr* out);
63 absl::optional<std::string> name, ImmediateTensorHandlePtr handle);
|
| D | restored_resource_revival_state.h | 33 ImmediateTensorHandlePtr resource_handle = nullptr;
|
| D | constant.h | 48 explicit Constant(ImmediateTensorHandlePtr handle);
|
| D | restored_resource.h | 68 ImmediateTensorHandlePtr resource_handle);
|
| D | asset.h | 43 explicit Asset(ImmediateTensorHandlePtr handle);
|
| D | restored_resource.cc | 50 ImmediateTensorHandlePtr resource_handle) in RestoredResource()
|
| D | partially_revived_objects.cc | 421 ImmediateTensorHandlePtr result( in CreateAllResourceHandles()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 71 ImmediateTensorHandlePtr CreateTensorHandle(ImmediateExecutionContext* ctx, in CreateTensorHandle()
85 ImmediateTensorHandlePtr handle(ctx->CreateLocalHandle(tensor.get())); in CreateTensorHandle()
|
| D | saved_variable_loading_test.cc | 125 ImmediateTensorHandlePtr expected_handle = in TEST_P()
134 ImmediateTensorHandlePtr output_handle; in TEST_P()
|
| D | test_utils.h | 55 ImmediateTensorHandlePtr CreateTensorHandle(ImmediateExecutionContext* ctx,
|
| D | tf_saved_model_api.cc | 153 ImmediateTensorHandlePtr restored_output; in RestoreCheckpoint()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_tensor_handle.h | 86 using ImmediateTensorHandlePtr = variable
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad.cc | 82 ImmediateTensorHandlePtr dim(imm_ctx->CreateLocalHandle(minus_1.get())); in BroadcastMul()
|