Home
last modified time | relevance | path

Searched refs:AbstractContext (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.h23 Status Mul(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
26 Status Conj(AbstractContext* ctx,
30 Status Add(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
33 Status MatMul(AbstractContext* ctx,
38 Status Neg(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
41 Status Sum(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
44 Status Sub(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
47 Status Div(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
50 Status DivNoNan(AbstractContext* ctx,
54 Status Exp(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
[all …]
Darray_ops.h25 Status Identity(AbstractContext* ctx,
29 Status IdentityN(AbstractContext* ctx,
33 Status ZerosLike(AbstractContext* ctx,
37 Status Shape(AbstractContext* ctx,
41 Status ExpandDims(AbstractContext* ctx,
45 Status OnesLike(AbstractContext* ctx,
Dmath_ops.cc29 Status Mul(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Mul()
40 Status Conj(AbstractContext* ctx, in Conj()
63 Status Add(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Add()
76 Status Sub(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Sub()
89 Status MatMul(AbstractContext* ctx, in MatMul()
107 Status Neg(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Neg()
118 Status Sum(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Sum()
131 Status Div(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Div()
144 Status DivNoNan(AbstractContext* ctx, in DivNoNan()
159 Status Exp(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in Exp()
[all …]
Dnn_ops.h26 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
29 Status ReluGrad(AbstractContext* ctx,
33 Status Relu(AbstractContext* ctx,
37 Status BiasAdd(AbstractContext* ctx,
41 Status BiasAddGrad(AbstractContext* ctx,
Darray_ops.cc25 Status Identity(AbstractContext* ctx, in Identity()
37 Status IdentityN(AbstractContext* ctx, in IdentityN()
49 Status ZerosLike(AbstractContext* ctx, in ZerosLike()
60 Status Shape(AbstractContext* ctx, in Shape()
72 Status ExpandDims(AbstractContext* ctx, in ExpandDims()
84 Status OnesLike(AbstractContext* ctx, in OnesLike()
Dnn_ops.cc28 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in SparseSoftmaxCrossEntropyWithLogits()
44 Status ReluGrad(AbstractContext* ctx, in ReluGrad()
59 Status Relu(AbstractContext* ctx, in Relu()
72 Status BiasAdd(AbstractContext* ctx, in BiasAdd()
88 Status BiasAddGrad(AbstractContext* ctx, in BiasAddGrad()
/external/tensorflow/tensorflow/c/eager/
Dgradients_util.h39 Status ScalarTensorHandle(AbstractContext* ctx, float value,
43 Status TensorHandleWithDimsFloat(AbstractContext* ctx, float data[],
48 Status TensorHandleWithDimsInt(AbstractContext* ctx, int data[], int64_t dims[],
55 AbstractTensorHandlePtr GetTensorHandleUtilFloat(AbstractContext* ctx,
60 AbstractTensorHandlePtr GetTensorHandleUtilInt(AbstractContext* ctx, int vals[],
64 AbstractTensorHandlePtr GetScalarTensorHandleUtil(AbstractContext* ctx,
68 Status UpdateWeights(AbstractContext* ctx,
74 AbstractContext*, absl::Span<AbstractTensorHandle* const>,
79 Status RunModel(Model model, AbstractContext* ctx,
85 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx);
Dunified_api_testutil.h26 AbstractContext* BuildFunction(const char* fn_name);
30 Status CreateParamsForInputs(AbstractContext* ctx,
35 using Model = std::function<Status(AbstractContext*,
46 Status RunModel(Model model, AbstractContext* ctx,
50 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx);
53 Status TestScalarTensorHandle(AbstractContext* ctx, float value,
57 Status TestTensorHandleWithDimsFloat(AbstractContext* ctx, float* data,
62 Status TestTensorHandleWithDimsInt(AbstractContext* ctx, int* data,
Dmnist_gradients_testutil.h39 Status AddGradModel(AbstractContext* ctx,
47 Status MatMulGradModel(AbstractContext* ctx,
53 Status MNISTForwardModel(AbstractContext* ctx,
59 Status MatMulTransposeModel(AbstractContext* ctx,
65 Status MNISTGradModel(AbstractContext* ctx,
71 Status ScalarMulModel(AbstractContext* ctx,
76 Status MatMulModel(AbstractContext* ctx,
81 Status MulModel(AbstractContext* ctx,
Dabstract_context.h33 class AbstractContext {
36 explicit AbstractContext(AbstractContextKind kind) : kind_(kind) {} in AbstractContext() function
37 virtual ~AbstractContext() {} in ~AbstractContext()
69 void operator()(AbstractContext* p) const { in operator()
78 std::unique_ptr<AbstractContext, internal::AbstractContextDeleter>;
Dmnist_gradients_testutil.cc45 Status AddGradModel(AbstractContext* ctx, in AddGradModel()
69 Status MatMulGradModel(AbstractContext* ctx, in MatMulGradModel()
94 Status MNISTForwardModel(AbstractContext* ctx, in MNISTForwardModel()
149 Status MatMulTransposeModel(AbstractContext* ctx, in MatMulTransposeModel()
162 Status MNISTGradModel(AbstractContext* ctx, in MNISTGradModel()
215 Status ScalarMulModel(AbstractContext* ctx, in ScalarMulModel()
223 Status MatMulModel(AbstractContext* ctx, in MatMulModel()
232 Status MulModel(AbstractContext* ctx, in MulModel()
Dgradients_util.cc83 Status ScalarTensorHandle(AbstractContext* ctx, float value, in ScalarTensorHandle()
98 Status TensorHandleWithDimsFloat(AbstractContext* ctx, float data[], in TensorHandleWithDimsFloat()
115 Status TensorHandleWithDimsInt(AbstractContext* ctx, int data[], int64_t dims[], in TensorHandleWithDimsInt()
140 AbstractTensorHandlePtr GetTensorHandleUtilFloat(AbstractContext* ctx, in GetTensorHandleUtilFloat()
152 AbstractTensorHandlePtr GetTensorHandleUtilInt(AbstractContext* ctx, int vals[], in GetTensorHandleUtilInt()
163 AbstractTensorHandlePtr GetScalarTensorHandleUtil(AbstractContext* ctx, in GetScalarTensorHandleUtil()
174 Status UpdateWeights(AbstractContext* ctx, vector<AbstractTensorHandle*>& grads, in UpdateWeights()
216 AbstractContext* BuildFunction(const char* fn_name) { in BuildFunction()
223 Status CreateParamsForInputs(AbstractContext* ctx, in CreateParamsForInputs()
237 Status RunModel(Model model, AbstractContext* ctx, in RunModel()
[all …]
Dunified_api_testutil.cc30 AbstractContext* BuildFunction(const char* fn_name) { in BuildFunction()
37 Status CreateParamsForInputs(AbstractContext* ctx, in CreateParamsForInputs()
52 Status RunModel(Model model, AbstractContext* ctx, in RunModel()
122 Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx) { in BuildImmediateExecutionContext()
133 Status TestScalarTensorHandle(AbstractContext* ctx, float value, in TestScalarTensorHandle()
146 Status TestTensorHandleWithDimsFloat(AbstractContext* ctx, float* data, in TestTensorHandleWithDimsFloat()
161 Status TestTensorHandleWithDimsInt(AbstractContext* ctx, int* data, in TestTensorHandleWithDimsInt()
Dc_api_unified_experimental_internal.h105 class TracingContext : public AbstractContext {
107 explicit TracingContext(AbstractContextKind kind) : AbstractContext(kind) {} in TracingContext()
119 static bool classof(const AbstractContext* ptr) { in classof()
130 DEFINE_CONVERSION_FUNCTIONS(AbstractContext, TF_ExecutionContext)
Dgradient_checker_test.cc34 Model model, AbstractContext* ctx, in CompareNumericalAndManualGradients()
65 Status MatMulModel(AbstractContext* ctx, in MatMulModel()
73 Status MulModel(AbstractContext* ctx, in MulModel()
94 AbstractContext* ctx_raw = nullptr; in SetUp()
/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.cc28 using tensorflow::AbstractContext;
33 m.def("add", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
44 m.def("mat_mul", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
57 [](AbstractContext* ctx, AbstractTensorHandle* a, const char* name) { in PYBIND11_MODULE()
67 m.def("sub", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
78 m.def("mul", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
90 [](AbstractContext* ctx, AbstractTensorHandle* a, const char* name) { in PYBIND11_MODULE()
100 m.def("div_no_nan", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
Dunified_api.cc45 using tensorflow::AbstractContext;
114 py::class_<AbstractContext, AbstractContextPtr>(m, "AbstractContext") in PYBIND11_MODULE()
116 [](AbstractContext* self, const char* op, in PYBIND11_MODULE()
123 [](AbstractContext* self, AbstractFunction* f) { in PYBIND11_MODULE()
127 .def("RemoveFunction", [](AbstractContext* self, const string& func) { in PYBIND11_MODULE()
132 py::class_<TracingContext, AbstractContext>(m, "TracingContext") in PYBIND11_MODULE()
179 py::class_<ImmediateExecutionContext, AbstractContext, in PYBIND11_MODULE()
Dnn_ops.cc28 using tensorflow::AbstractContext;
34 [](AbstractContext* ctx, AbstractTensorHandle* a, const char* name) { in PYBIND11_MODULE()
47 [](AbstractContext* ctx, AbstractTensorHandle* features, in PYBIND11_MODULE()
Dtape.cc52 [](Tape* self, AbstractContext* ctx, in PYBIND11_MODULE()
68 py::class_<TapeContext, AbstractContext>(m, "TapeContext") in PYBIND11_MODULE()
70 [](AbstractContext* ctx, Tape* tape, GradientRegistry* registry) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_context.h23 class TapeContext : public AbstractContext {
25 explicit TapeContext(AbstractContext*, Tape*, const GradientRegistry&);
31 static bool classof(const AbstractContext* ptr) { in classof()
37 AbstractContext* parent_ctx_; // Not owned.
/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal.h87 class OpHandlerContext : public AbstractContext {
89 explicit OpHandlerContext(AbstractContext*);
95 static bool classof(const AbstractContext* ptr) { in classof()
103 AbstractContext* parent_ctx_; // Not owned.
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad_test.cc34 Status AddModel(AbstractContext* ctx, in AddModel()
40 Status ExpModel(AbstractContext* ctx, in ExpModel()
46 Status SqrtModel(AbstractContext* ctx, in SqrtModel()
52 Status NegModel(AbstractContext* ctx, in NegModel()
58 Status SubModel(AbstractContext* ctx, in SubModel()
64 Status MulModel(AbstractContext* ctx, in MulModel()
70 Status Log1pModel(AbstractContext* ctx, in Log1pModel()
76 Status DivNoNanModel(AbstractContext* ctx, in DivNoNanModel()
92 AbstractContext* ctx_raw = nullptr; in SetUp()
194 AbstractContext* ctx, in TEST_P()
Dmath_grad.cc40 Status Compute(AbstractContext* ctx, in Compute()
61 Status Compute(AbstractContext* ctx, in Compute()
86 Status Compute(AbstractContext* ctx, in Compute()
112 Status Compute(AbstractContext* ctx, in Compute()
224 Status Compute(AbstractContext* ctx, in Compute()
243 Status Compute(AbstractContext* ctx, in Compute()
280 Status Compute(AbstractContext* ctx, in Compute()
327 Status Compute(AbstractContext* ctx, in Compute()
400 Status Compute(AbstractContext* ctx, in Compute()
Dnn_grad_test.cc32 Status ReluModel(AbstractContext* ctx, in ReluModel()
38 Status ReluGradModel(AbstractContext* ctx, in ReluGradModel()
61 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in SparseSoftmaxCrossEntropyWithLogitsModel()
77 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in SparseSoftmaxCrossEntropyWithLogitsGradModel()
102 Status BiasAddModel(AbstractContext* ctx, in BiasAddModel()
108 Status BiasAddGradModel(AbstractContext* ctx, in BiasAddGradModel()
141 AbstractContext* ctx_raw = nullptr; in SetUp()
Dnn_grad.cc47 Status Compute(AbstractContext* ctx, in Compute()
72 Status BroadcastMul(AbstractContext* ctx, AbstractTensorHandle* vec, in BroadcastMul()
100 Status Compute(AbstractContext* ctx, in Compute()
124 Status Compute(AbstractContext* ctx, in Compute()

12