Home
last modified time | relevance | path

Searched refs:AbstractTensorHandle (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.h28 Status Mul(AbstractContext* ctx, AbstractTensorHandle* const x,
29 AbstractTensorHandle* const y, AbstractTensorHandle** z,
33 Status Conj(AbstractContext* ctx, AbstractTensorHandle* const input,
34 AbstractTensorHandle** output, const char* name = nullptr,
38 Status AddV2(AbstractContext* ctx, AbstractTensorHandle* const x,
39 AbstractTensorHandle* const y, AbstractTensorHandle** z,
43 Status MatMul(AbstractContext* ctx, AbstractTensorHandle* const a,
44 AbstractTensorHandle* const b, AbstractTensorHandle** product,
50 Status Neg(AbstractContext* ctx, AbstractTensorHandle* const x,
51 AbstractTensorHandle** y, const char* name = nullptr,
[all …]
Dmath_ops.cc37 Status Mul(AbstractContext* ctx, AbstractTensorHandle* const x, in Mul()
38 AbstractTensorHandle* const y, AbstractTensorHandle** z, in Mul()
67 Status Conj(AbstractContext* ctx, AbstractTensorHandle* const input, in Conj()
68 AbstractTensorHandle** output, const char* name, in Conj()
85 Status AddV2(AbstractContext* ctx, AbstractTensorHandle* const x, in AddV2()
86 AbstractTensorHandle* const y, AbstractTensorHandle** z, in AddV2()
108 Status MatMul(AbstractContext* ctx, AbstractTensorHandle* const a, in MatMul()
109 AbstractTensorHandle* const b, AbstractTensorHandle** product, in MatMul()
128 Status Neg(AbstractContext* ctx, AbstractTensorHandle* const x, in Neg()
129 AbstractTensorHandle** y, const char* name, in Neg()
[all …]
Dnn_ops.h29 AbstractContext* ctx, AbstractTensorHandle* const features,
30 AbstractTensorHandle* const labels, AbstractTensorHandle** loss,
31 AbstractTensorHandle** backprop, const char* name = nullptr,
35 Status ReluGrad(AbstractContext* ctx, AbstractTensorHandle* const gradients,
36 AbstractTensorHandle* const features,
37 AbstractTensorHandle** backprops, const char* name = nullptr,
41 Status Relu(AbstractContext* ctx, AbstractTensorHandle* const features,
42 AbstractTensorHandle** activations, const char* name = nullptr,
46 Status BiasAdd(AbstractContext* ctx, AbstractTensorHandle* const value,
47 AbstractTensorHandle* const bias, AbstractTensorHandle** output,
[all …]
Darray_ops.h29 Status Identity(AbstractContext* ctx, AbstractTensorHandle* const input,
30 AbstractTensorHandle** output, const char* name = nullptr,
35 absl::Span<AbstractTensorHandle* const> input,
36 absl::Span<AbstractTensorHandle*> output,
41 Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x,
42 AbstractTensorHandle** y, const char* name = nullptr,
46 Status Shape(AbstractContext* ctx, AbstractTensorHandle* const input,
47 AbstractTensorHandle** output, DataType out_type = DT_INT32,
51 Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input,
52 AbstractTensorHandle* const dim,
[all …]
Dnn_ops.cc42 AbstractTensorHandle* const features, in SparseSoftmaxCrossEntropyWithLogits()
43 AbstractTensorHandle* const labels, in SparseSoftmaxCrossEntropyWithLogits()
44 AbstractTensorHandle** loss, in SparseSoftmaxCrossEntropyWithLogits()
45 AbstractTensorHandle** backprop, in SparseSoftmaxCrossEntropyWithLogits()
55 AbstractTensorHandle* temp_outputs[2]; in SparseSoftmaxCrossEntropyWithLogits()
66 Status ReluGrad(AbstractContext* ctx, AbstractTensorHandle* const gradients, in ReluGrad()
67 AbstractTensorHandle* const features, in ReluGrad()
68 AbstractTensorHandle** backprops, const char* name, in ReluGrad()
87 Status Relu(AbstractContext* ctx, AbstractTensorHandle* const features, in Relu()
88 AbstractTensorHandle** activations, const char* name, in Relu()
[all …]
Darray_ops.cc36 Status Identity(AbstractContext* ctx, AbstractTensorHandle* const input, in Identity()
37 AbstractTensorHandle** output, const char* name, in Identity()
68 absl::Span<AbstractTensorHandle* const> input, in IdentityN()
69 absl::Span<AbstractTensorHandle*> output, const char* name, in IdentityN()
83 Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x, in ZerosLike()
84 AbstractTensorHandle** y, const char* name, in ZerosLike()
107 Status Shape(AbstractContext* ctx, AbstractTensorHandle* const input, in Shape()
108 AbstractTensorHandle** output, DataType out_type, const char* name, in Shape()
153 Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input, in ExpandDims()
154 AbstractTensorHandle* const dim, in ExpandDims()
[all …]
Dio_ops.h28 Status RestoreV2(AbstractContext* ctx, AbstractTensorHandle* const prefix,
29 AbstractTensorHandle* const tensor_names,
30 AbstractTensorHandle* const shape_and_slices,
31 absl::Span<AbstractTensorHandle*> tensors,
36 Status SaveV2(AbstractContext* ctx, AbstractTensorHandle* const prefix,
37 AbstractTensorHandle* const tensor_names,
38 AbstractTensorHandle* const shape_and_slices,
39 absl::Span<AbstractTensorHandle* const> tensors,
Dio_ops.cc50 Status RestoreV2(AbstractContext* ctx, AbstractTensorHandle* const prefix, in RestoreV2()
51 AbstractTensorHandle* const tensor_names, in RestoreV2()
52 AbstractTensorHandle* const shape_and_slices, in RestoreV2()
53 absl::Span<AbstractTensorHandle*> tensors, in RestoreV2()
75 Status SaveV2(AbstractContext* ctx, AbstractTensorHandle* const prefix, in SaveV2()
76 AbstractTensorHandle* const tensor_names, in SaveV2()
77 AbstractTensorHandle* const shape_and_slices, in SaveV2()
78 absl::Span<AbstractTensorHandle* const> tensors, const char* name, in SaveV2()
88 std::vector<AbstractTensorHandle*> dummy_outputs; in SaveV2()
Dresource_variable_ops.h28 Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource,
37 AbstractTensorHandle* const resource,
38 AbstractTensorHandle** value, DataType dtype,
44 AbstractTensorHandle* const resource,
45 AbstractTensorHandle* const value,
51 AbstractTensorHandle* const resource,
Dresource_variable_ops.cc35 Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource, in VarHandleOp()
66 AbstractTensorHandle* const resource, in ReadVariableOp()
67 AbstractTensorHandle** value, DataType dtype, in ReadVariableOp()
85 AbstractTensorHandle* const resource, in AssignVariableOp()
86 AbstractTensorHandle* const value, bool validate_shape, in AssignVariableOp()
95 std::vector<AbstractTensorHandle*> dummy_outputs; in AssignVariableOp()
106 AbstractTensorHandle* const resource, in DestroyResourceOp()
116 std::vector<AbstractTensorHandle*> dummy_outputs; in DestroyResourceOp()
/external/tensorflow/tensorflow/python/framework/experimental/
Dmath_ops.cc29 using tensorflow::AbstractTensorHandle;
33 m.def("add", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
34 AbstractTensorHandle* b, const char* name) { in PYBIND11_MODULE()
36 AbstractTensorHandle* output; in PYBIND11_MODULE()
43 m.def("mat_mul", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
44 AbstractTensorHandle* b, const char* name) { in PYBIND11_MODULE()
46 AbstractTensorHandle* output; in PYBIND11_MODULE()
56 [](AbstractContext* ctx, AbstractTensorHandle* a, const char* name) { in PYBIND11_MODULE()
58 AbstractTensorHandle* output; in PYBIND11_MODULE()
65 m.def("sub", [](AbstractContext* ctx, AbstractTensorHandle* a, in PYBIND11_MODULE()
[all …]
Dnn_ops.cc29 using tensorflow::AbstractTensorHandle;
34 [](AbstractContext* ctx, AbstractTensorHandle* a, const char* name) { in PYBIND11_MODULE()
36 AbstractTensorHandle* output; in PYBIND11_MODULE()
46 [](AbstractContext* ctx, AbstractTensorHandle* features, in PYBIND11_MODULE()
47 AbstractTensorHandle* labels, const char* name) { in PYBIND11_MODULE()
49 AbstractTensorHandle* loss; in PYBIND11_MODULE()
50 AbstractTensorHandle* backprop; in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/eager/
Dgradients.h58 absl::Span<AbstractTensorHandle* const> grad_outputs,
59 absl::Span<AbstractTensorHandle*> grad_inputs) = 0;
68 std::vector<AbstractTensorHandle*> inputs;
69 std::vector<AbstractTensorHandle*> outputs;
110 explicit TapeTensor(AbstractTensorHandle* handle);
117 AbstractTensorHandle* ZerosLike() const;
119 AbstractTensorHandle* GetHandle() const;
122 AbstractTensorHandle* handle_;
129 class Tape : protected eager::GradientTape<AbstractTensorHandle,
132 using GradientTape<AbstractTensorHandle, GradientFunction,
[all …]
Dgradients.cc32 int64_t ToId(const AbstractTensorHandle* t) { in ToId()
36 Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* t, in ZerosLike()
37 AbstractTensorHandle** result) { in ZerosLike()
46 std::vector<AbstractTensorHandle*> outputs(num_outputs); in ZerosLike()
48 op->Execute(absl::Span<AbstractTensorHandle*>(outputs), &num_outputs)); in ZerosLike()
76 TapeTensor::TapeTensor(AbstractTensorHandle* handle) : handle_(handle) { in TapeTensor()
90 AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; } in GetHandle()
92 AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; } in ZerosLike()
95 : public eager::VSpace<AbstractTensorHandle, GradientFunction, TapeTensor> {
101 int64_t NumElements(AbstractTensorHandle* tensor) const override;
[all …]
Dunified_api_testutil.h35 absl::Span<AbstractTensorHandle* const> inputs,
36 std::vector<AbstractTensorHandle*>* params);
40 absl::Span<AbstractTensorHandle* const>,
41 absl::Span<AbstractTensorHandle*>)>;
51 absl::Span<AbstractTensorHandle* const> inputs,
52 absl::Span<AbstractTensorHandle*> outputs, bool use_function);
60 AbstractTensorHandle** tensor) { in TestTensorHandleWithDims()
76 AbstractTensorHandle** tensor) { in TestScalarTensorHandle()
90 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor);
Dgradient_checker.cc50 absl::Span<AbstractTensorHandle* const> inputs, in RunAndMaybeSum()
51 absl::Span<AbstractTensorHandle*> outputs, in RunAndMaybeSum()
53 AbstractTensorHandle* model_outputs[1]; in RunAndMaybeSum()
79 AbstractTensorHandle* sum_dims_raw = nullptr; in RunAndMaybeSum()
93 absl::Span<AbstractTensorHandle* const> inputs, in CalcNumericalGrad()
95 AbstractTensorHandle** numerical_grad) { in CalcNumericalGrad()
96 vector<AbstractTensorHandle*> theta_inputs(inputs.size()); in CalcNumericalGrad()
101 AbstractTensorHandle* theta = in CalcNumericalGrad()
125 AbstractTensorHandle* f_outputs[1]; in CalcNumericalGrad()
133 AbstractTensorHandle* two_eps_raw = nullptr; in CalcNumericalGrad()
[all …]
Dgradient_checker_test.cc35 absl::Span<AbstractTensorHandle* const> inputs, int input_index, in CompareNumericalAndManualGradients()
41 AbstractTensorHandle* numerical_grad_raw; in CompareNumericalAndManualGradients()
66 absl::Span<AbstractTensorHandle* const> inputs, in MatMulModel()
67 absl::Span<AbstractTensorHandle*> outputs) { in MatMulModel()
74 absl::Span<AbstractTensorHandle* const> inputs, in MulModel()
75 absl::Span<AbstractTensorHandle*> outputs) { in MulModel()
122 AbstractTensorHandle* A_raw; in TEST_P()
132 AbstractTensorHandle* B_raw; in TEST_P()
148 AbstractTensorHandle* x_raw = nullptr; in TEST_P()
157 AbstractTensorHandle* y_raw = nullptr; in TEST_P()
Dunified_api_testutil.cc38 absl::Span<AbstractTensorHandle* const> inputs, in CreateParamsForInputs()
39 std::vector<AbstractTensorHandle*>* params) { in CreateParamsForInputs()
53 absl::Span<AbstractTensorHandle* const> inputs, in RunModel()
54 absl::Span<AbstractTensorHandle*> outputs, bool use_function) { in RunModel()
65 std::vector<AbstractTensorHandle*> func_inputs; in RunModel()
69 std::vector<AbstractTensorHandle*> model_outputs; in RunModel()
103 std::vector<AbstractTensorHandle*> fn_outputs(retvals); in RunModel()
105 absl::Span<AbstractTensorHandle*>(fn_outputs.data(), fn_outputs.size()), in RunModel()
133 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { in GetValue()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad.cc37 static Status SafeConj(AbstractContext* ctx, AbstractTensorHandle* const input, in SafeConj()
38 AbstractTensorHandle** output, const char* name) { in SafeConj()
54 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
55 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
71 explicit ExpGradientFunction(AbstractTensorHandle* exp) : exp_(exp) { in ExpGradientFunction()
75 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
76 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
77 AbstractTensorHandle* conj_output; in Compute()
95 explicit SqrtGradientFunction(AbstractTensorHandle* sqrt) : sqrt_(sqrt) { in SqrtGradientFunction()
99 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
[all …]
Dnn_grad.cc38 explicit ReluGradientFunction(vector<AbstractTensorHandle*> f_outputs) in ReluGradientFunction()
48 absl::Span<AbstractTensorHandle* const> grad_outputs, in Compute()
49 absl::Span<AbstractTensorHandle*> grad_inputs) override { in Compute()
50 AbstractTensorHandle* upstream_grad = grad_outputs[0]; in Compute()
51 AbstractTensorHandle* activations = forward_outputs_[0]; in Compute()
69 vector<AbstractTensorHandle*> forward_outputs_;
72 Status BroadcastMul(AbstractContext* ctx, AbstractTensorHandle* vec, in BroadcastMul()
73 AbstractTensorHandle* mat, in BroadcastMul()
74 absl::Span<AbstractTensorHandle*> outputs) { in BroadcastMul()
83 AbstractTensorHandle* expand_dims_outputs; in BroadcastMul()
[all …]
Dmath_grad_test.cc35 absl::Span<AbstractTensorHandle* const> inputs, in AddModel()
36 absl::Span<AbstractTensorHandle*> outputs) { in AddModel()
41 absl::Span<AbstractTensorHandle* const> inputs, in ExpModel()
42 absl::Span<AbstractTensorHandle*> outputs) { in ExpModel()
47 absl::Span<AbstractTensorHandle* const> inputs, in SqrtModel()
48 absl::Span<AbstractTensorHandle*> outputs) { in SqrtModel()
53 absl::Span<AbstractTensorHandle* const> inputs, in NegModel()
54 absl::Span<AbstractTensorHandle*> outputs) { in NegModel()
59 absl::Span<AbstractTensorHandle* const> inputs, in SubModel()
60 absl::Span<AbstractTensorHandle*> outputs) { in SubModel()
[all …]
Dnn_grad_test.cc34 absl::Span<AbstractTensorHandle* const> inputs, in ReluModel()
35 absl::Span<AbstractTensorHandle*> outputs) { in ReluModel()
40 AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, in SparseSoftmaxCrossEntropyWithLogitsModel()
41 absl::Span<AbstractTensorHandle*> outputs) { in SparseSoftmaxCrossEntropyWithLogitsModel()
42 AbstractTensorHandle* loss; in SparseSoftmaxCrossEntropyWithLogitsModel()
43 AbstractTensorHandle* backprop; in SparseSoftmaxCrossEntropyWithLogitsModel()
57 absl::Span<AbstractTensorHandle* const> inputs, in BiasAddModel()
58 absl::Span<AbstractTensorHandle*> outputs) { in BiasAddModel()
105 AbstractTensorHandle* X_raw; in TEST_P()
120 AbstractTensorHandle* Y_raw; in TEST_P()
[all …]
/external/tensorflow/tensorflow/c/experimental/ops/gen/cpp/golden/
Dtesting_ops.h.golden28 Status Neg(AbstractContext* ctx, AbstractTensorHandle* const x, AbstractTensorHandle** y, const cha…
31 …tatus MatMul(AbstractContext* ctx, AbstractTensorHandle* const a, AbstractTensorHandle* const b, A…
34 …IdentityN(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> input, absl::Span<Abstract…
37 …tContext* ctx, AbstractTensorHandle* const features, AbstractTensorHandle* const labels, AbstractT…
40 …Gradient(AbstractContext* ctx, AbstractTensorHandle** handle, AbstractTensorHandle* const local_st…
43 Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource, DataType dtype, const Par…
46 …AbstractTensorHandle* const prefix, AbstractTensorHandle* const tensor_names, AbstractTensorHandle…
Dtesting_ops.cc.golden35 Status Neg(AbstractContext* ctx, AbstractTensorHandle* const x, AbstractTensorHandle** y, const cha…
48 …tatus MatMul(AbstractContext* ctx, AbstractTensorHandle* const a, AbstractTensorHandle* const b, A…
64 …IdentityN(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> input, absl::Span<Abstract…
77 …tContext* ctx, AbstractTensorHandle* const features, AbstractTensorHandle* const labels, AbstractT…
84 AbstractTensorHandle* temp_outputs[2];
95 …Gradient(AbstractContext* ctx, AbstractTensorHandle** handle, AbstractTensorHandle* const local_st…
103 std::vector<AbstractTensorHandle*> dummy_outputs;
111 Status VarHandleOp(AbstractContext* ctx, AbstractTensorHandle** resource, DataType dtype, const Par…
128 …AbstractTensorHandle* const prefix, AbstractTensorHandle* const tensor_names, AbstractTensorHandle…
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache_test.cc61 absl::Span<tensorflow::AbstractTensorHandle* const> inputs, in Add()
62 absl::Span<tensorflow::AbstractTensorHandle*> outputs) { in Add()
83 absl::Span<tensorflow::AbstractTensorHandle* const> inputs, in AddModel()
84 absl::Span<tensorflow::AbstractTensorHandle*> outputs) { in AddModel()
85 std::vector<tensorflow::AbstractTensorHandle*> add_outputs(1); in AddModel()
102 absl::Span<tensorflow::AbstractTensorHandle* const> inputs, in CreateParamsForInputs()
103 std::vector<tensorflow::AbstractTensorHandle*>* params) { in CreateParamsForInputs()
118 absl::Span<tensorflow::AbstractTensorHandle* const>,
119 absl::Span<tensorflow::AbstractTensorHandle*>)>;
123 absl::Span<tensorflow::AbstractTensorHandle* const> inputs, in PrepareFunction()
[all …]

1234