| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.h | 49 Status TensorProtoToConstant(ImmediateExecutionContext* ctx,
57 Status LoadSavedVariable(ImmediateExecutionContext* ctx,
61 Status LoadSavedAsset(ImmediateExecutionContext* ctx, const SavedAsset& asset,
72 ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out);
101 ImmediateExecutionContext* context,
|
| D | saved_model_utils.cc | 63 ImmediateExecutionContext* ctx, in ConstantFromSavedConstant()
203 Status LoadSavedAsset(ImmediateExecutionContext* ctx, const SavedAsset& asset, in LoadSavedAsset()
217 Status TensorProtoToConstant(ImmediateExecutionContext* ctx, in TensorProtoToConstant()
232 Status LoadSavedVariable(ImmediateExecutionContext* ctx, in LoadSavedVariable()
257 ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out) { in LoadTFConcreteFunction()
384 ImmediateExecutionContext* context, in PartiallyReviveSavedModelObjects()
|
| D | test_utils.h | 55 ImmediateTensorHandlePtr CreateTensorHandle(ImmediateExecutionContext* ctx,
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops.h | 32 Status CreateUninitializedResourceVariable(ImmediateExecutionContext* ctx,
42 Status AssignVariable(ImmediateExecutionContext* ctx,
49 Status ReadVariable(ImmediateExecutionContext* ctx,
56 Status DestroyResource(ImmediateExecutionContext* ctx,
|
| D | variable_ops.cc | 38 Status CreateUninitializedResourceVariable(ImmediateExecutionContext* ctx, in CreateUninitializedResourceVariable()
71 Status AssignVariable(ImmediateExecutionContext* ctx, in AssignVariable()
85 Status ReadVariable(ImmediateExecutionContext* ctx, in ReadVariable()
106 Status DestroyResource(ImmediateExecutionContext* ctx, in DestroyResource()
|
| D | restore_ops.cc | 35 Status CreateStringScalarTensorHandle(ImmediateExecutionContext* ctx, in CreateStringScalarTensorHandle()
49 Status CreateStringVectorTensorHandle(ImmediateExecutionContext* ctx, in CreateStringVectorTensorHandle()
69 Status SingleRestore(ImmediateExecutionContext* ctx, const std::string& prefix, in SingleRestore()
|
| D | restore_ops.h | 33 Status SingleRestore(ImmediateExecutionContext* ctx, const std::string& prefix,
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | flat_tensor_function.h | 61 ImmediateExecutionContext* ctx,
73 ImmediateExecutionContext* ctx);
81 ImmediateExecutionContext* ctx_;
|
| D | variable.h | 38 ImmediateExecutionContext* ctx, DataType dtype, TensorShape shape,
62 Variable(ImmediateExecutionContext* ctx, DataType dtype, TensorShape shape,
72 ImmediateExecutionContext* ctx_;
|
| D | partially_revived_objects.cc | 279 Status CreateConcreteFunction(ImmediateExecutionContext* ctx, in CreateConcreteFunction()
302 ImmediateExecutionContext* ctx, in CreateSignatureDefFunction()
327 Status InitializeCreateResourceFunctions(ImmediateExecutionContext* ctx, in InitializeCreateResourceFunctions()
352 Status InitializeAllFunctions(ImmediateExecutionContext* ctx, in InitializeAllFunctions()
393 Status CreateAllResourceHandles(ImmediateExecutionContext* ctx, in CreateAllResourceHandles()
429 Status BuildResources(ImmediateExecutionContext* ctx, in BuildResources()
493 Status PartiallyRevivedObjects::Build(ImmediateExecutionContext* ctx, in Build()
|
| D | flat_tensor_function.cc | 37 ImmediateExecutionContext* ctx) in FlatTensorFunction()
51 ImmediateExecutionContext* ctx, std::unique_ptr<FlatTensorFunction>* out) { in Create()
|
| D | variable.cc | 32 Variable::Variable(ImmediateExecutionContext* ctx, DataType dtype, in Variable()
68 ImmediateExecutionContext* ctx, DataType dtype, TensorShape shape, in CreateUninitialized()
|
| D | constant.cc | 35 Status Constant::Create(ImmediateExecutionContext* ctx, in Create()
|
| D | constant.h | 37 static Status Create(ImmediateExecutionContext* ctx,
|
| D | asset.h | 31 static Status Create(ImmediateExecutionContext* ctx,
|
| D | partially_revived_objects.h | 56 Status Build(ImmediateExecutionContext* ctx,
|
| D | asset.cc | 32 Status Asset::Create(ImmediateExecutionContext* ctx, in Create()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 64 class ImmediateExecutionContext : public AbstractContext {
223 explicit ImmediateExecutionContext(AbstractContextKind kind) in ImmediateExecutionContext() function
225 ~ImmediateExecutionContext() override {} in ~ImmediateExecutionContext()
230 void operator()(ImmediateExecutionContext* p) const { in operator()
239 std::unique_ptr<ImmediateExecutionContext,
|
| D | c_api_unified_experimental_eager.cc | 39 using tensorflow::ImmediateExecutionContext;
74 auto imm_ctx = dyn_cast<ImmediateExecutionContext>(unwrap(ctx)); in TF_ExecutionContextGetTFEContext()
|
| D | immediate_execution_operation.h | 36 class ImmediateExecutionContext; variable
50 virtual ImmediateExecutionContext* GetContext() const = 0;
|
| D | tfe_context_internal.h | 31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionContext, TFE_Context);
|
| D | immediate_execution_distributed_manager.h | 22 class ImmediateExecutionContext; variable
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 59 using tensorflow::ImmediateExecutionContext;
179 py::class_<ImmediateExecutionContext, AbstractContext, in PYBIND11_MODULE()
180 std::unique_ptr<ImmediateExecutionContext, py::nodelete>> in PYBIND11_MODULE()
181 ImmediateExecutionContext(m, "ImmediateExecutionContext"); in PYBIND11_MODULE() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.h | 74 CustomDeviceTensorHandle(ImmediateExecutionContext* context, in CustomDeviceTensorHandle()
115 ImmediateExecutionContext* const context_;
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | nn_grad.cc | 75 if (!isa<ImmediateExecutionContext>(ctx)) { in BroadcastMul()
80 auto imm_ctx = dyn_cast<ImmediateExecutionContext>(ctx); in BroadcastMul()
|