Home
last modified time | relevance | path

Searched refs:TensorRef (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dresize_test.cc32 TensorRef<BHWC> input; in TEST()
37 TensorRef<BHWC> output; in TEST()
57 TensorRef<BHWC> input; in TEST()
62 TensorRef<BHWC> output; in TEST()
81 TensorRef<BHWC> input; in TEST()
86 TensorRef<BHWC> output; in TEST()
107 TensorRef<BHWC> input; in TEST()
112 TensorRef<BHWC> output; in TEST()
133 TensorRef<BHWC> input; in TEST()
138 TensorRef<BHWC> output; in TEST()
[all …]
Dmul_test.cc34 TensorRef<BHWC> input; in TEST()
39 TensorRef<BHWC> output; in TEST()
54 TensorRef<BHWC> input; in TEST()
59 TensorRef<BHWC> output; in TEST()
78 TensorRef<BHWC> input; in TEST()
83 TensorRef<BHWC> output; in TEST()
104 TensorRef<BHWC> input; in TEST()
109 TensorRef<BHWC> mask; in TEST()
114 TensorRef<BHWC> output; in TEST()
128 TensorRef<BHWC> input; in TEST()
[all …]
Dslice_test.cc34 TensorRef<BHWC> input; in TEST()
39 TensorRef<BHWC> output; in TEST()
57 TensorRef<BHWC> input; in TEST()
62 TensorRef<BHWC> output; in TEST()
80 TensorRef<BHWC> input; in TEST()
85 TensorRef<BHWC> output; in TEST()
103 TensorRef<BHWC> input; in TEST()
108 TensorRef<BHWC> output; in TEST()
126 TensorRef<BHWC> input; in TEST()
131 TensorRef<BHWC> output; in TEST()
[all …]
Dspace_to_depth_test.cc34 const TensorRef<BHWC> input = { in TEST()
36 const TensorRef<BHWC> output = { in TEST()
48 const TensorRef<BHWC> input = { in TEST()
50 const TensorRef<BHWC> output = { in TEST()
64 const TensorRef<BHWC> input = { in TEST()
66 const TensorRef<BHWC> output = { in TEST()
83 const TensorRef<BHWC> input = { in TEST()
85 const TensorRef<BHWC> output = { in TEST()
Dprelu_test.cc32 TensorRef<BHWC> input; in TEST()
45 TensorRef<BHWC> output; in TEST()
58 TensorRef<BHWC> input; in TEST()
71 TensorRef<BHWC> output; in TEST()
84 TensorRef<BHWC> input;
98 TensorRef<BHWC> output;
110 TensorRef<BHWC> input;
124 TensorRef<BHWC> output;
136 TensorRef<BHWC> input;
150 TensorRef<BHWC> output;
[all …]
Dsoftmax_test.cc36 TensorRef<BHWC> input; in TEST()
41 TensorRef<BHWC> output; in TEST()
58 TensorRef<BHWC> input; in TEST()
63 TensorRef<BHWC> output; in TEST()
78 TensorRef<BHWC> input; in TEST()
83 TensorRef<BHWC> output; in TEST()
98 TensorRef<BHWC> input; in TEST()
103 TensorRef<BHWC> output; in TEST()
125 TensorRef<BHWC> input; in TEST()
130 TensorRef<BHWC> output; in TEST()
[all …]
Dreshape_test.cc34 TensorRef<BHWC> input;
39 TensorRef<BHWC> output;
56 TensorRef<BHWC> input;
61 TensorRef<BHWC> output;
78 TensorRef<BHWC> input;
83 TensorRef<BHWC> output;
99 TensorRef<BHWC> input; in TEST()
104 TensorRef<BHWC> output; in TEST()
Dquantize_and_dequantize_test.cc33 TensorRef<BHWC> input; in TEST()
48 TensorRef<BHWC> output; in TEST()
64 TensorRef<BHWC> input; in TEST()
79 TensorRef<BHWC> output; in TEST()
95 TensorRef<BHWC> input; in TEST()
110 TensorRef<BHWC> output; in TEST()
126 TensorRef<BHWC> input; in TEST()
141 TensorRef<BHWC> output; in TEST()
Dpooling_test.cc38 TensorRef<BHWC> input; in TEST()
43 TensorRef<BHWC> output; in TEST()
48 TensorRef<BHWC> indices; in TEST()
73 TensorRef<BHWC> input; in TEST()
78 TensorRef<BHWC> output; in TEST()
99 TensorRef<BHWC> input; in TEST()
104 TensorRef<BHWC> output; in TEST()
Dadd_test.cc34 TensorRef<BHWC> augend, addend, output; in TEST()
60 TensorRef<BHWC> input, output; in TEST()
78 TensorRef<BHWC> input; in TEST()
91 TensorRef<BHWC> output; in TEST()
107 TensorRef<BHWC> input1; in TEST()
112 TensorRef<BHWC> input2; in TEST()
119 TensorRef<BHWC> output; in TEST()
Dconv_test.cc34 TensorRef<BHWC> input; in TEST()
57 TensorRef<BHWC> output; in TEST()
72 TensorRef<BHWC> input; in TEST()
95 TensorRef<BHWC> output; in TEST()
109 TensorRef<BHWC> input; in TEST()
132 TensorRef<BHWC> output; in TEST()
146 TensorRef<BHWC> input; in TEST()
169 TensorRef<BHWC> output; in TEST()
183 TensorRef<BHWC> input; in TEST()
207 TensorRef<BHWC> output; in TEST()
Dtranspose_conv_test.cc34 TensorRef<BHWC> input; in TEST()
57 TensorRef<BHWC> output; in TEST()
73 TensorRef<BHWC> input; in TEST()
96 TensorRef<BHWC> output; in TEST()
114 TensorRef<BHWC> input; in TEST()
137 TensorRef<BHWC> output; in TEST()
153 TensorRef<BHWC> input; in TEST()
176 TensorRef<BHWC> output; in TEST()
191 TensorRef<BHWC> input; in TEST()
215 TensorRef<BHWC> output; in TEST()
Dlstm_test.cc34 TensorRef<BHWC> input; in TEST()
39 TensorRef<BHWC> prev_state; in TEST()
44 TensorRef<BHWC> output_state; in TEST()
49 TensorRef<BHWC> output_activation; in TEST()
Ddepthwise_conv_test.cc34 TensorRef<BHWC> input; in TEST()
58 TensorRef<BHWC> output; in TEST()
72 TensorRef<BHWC> input; in TEST()
96 TensorRef<BHWC> output; in TEST()
111 TensorRef<BHWC> input; in TEST()
135 TensorRef<BHWC> output; in TEST()
Dpad_test.cc37 TensorRef<BHWC> input; in TestPadOperation()
42 TensorRef<BHWC> output; in TestPadOperation()
123 TensorRef<BHWC> input; in TEST()
128 TensorRef<BHWC> output; in TEST()
146 TensorRef<BHWC> input; in TEST()
151 TensorRef<BHWC> output; in TEST()
Dmax_unpooling_test.cc34 TensorRef<BHWC> input; in TEST()
39 TensorRef<BHWC> indices; in TEST()
44 TensorRef<BHWC> output; in TEST()
Dconcat_test.cc34 TensorRef<BHWC> input1, input2, output; in TEST()
60 TensorRef<BHWC> input1, input2, output; in TEST()
86 TensorRef<BHWC> input1, input2, output; in TEST()
112 TensorRef<BHWC> input1, input2, output; in TEST()
Dmean_test.cc32 TensorRef<BHWC> input; in TEST()
37 TensorRef<BHWC> output; in TEST()
Dtest_util.h44 const std::vector<TensorRef<BHWC>>& inputs,
45 const std::vector<TensorRef<BHWC>>& outputs);
Dfully_connected_test.cc34 TensorRef<BHWC> input; in TEST()
53 TensorRef<BHWC> output; in TEST()
Drelu_test.cc36 TensorRef<BHWC> GetTensorRef(int ref) { in GetTensorRef()
37 TensorRef<BHWC> tensor_ref; in GetTensorRef()
Dtest_util.cc40 const std::vector<TensorRef<BHWC>>& inputs, in SingleOpModel()
41 const std::vector<TensorRef<BHWC>>& outputs) { in SingleOpModel()
/external/eigen/unsupported/test/
Dcxx11_tensor_ref.cpp22 TensorRef<Tensor<int, 1>> ref3(input); in test_simple_lvalue_ref()
23 TensorRef<Tensor<int, 1>> ref4 = input; in test_simple_lvalue_ref()
55 TensorRef<Tensor<int, 1>> ref3(input1 + input2); in test_simple_rvalue_ref()
56 TensorRef<Tensor<int, 1>> ref4 = input1 + input2; in test_simple_rvalue_ref()
75 TensorRef<Tensor<float, 3>> ref(input); in test_multiple_dims()
98 TensorRef<Tensor<float, 5>> slice = tensor.slice(indices, sizes); in test_slice()
124 TensorRef<Tensor<float, 3>> ref(input); in test_ref_of_ref()
125 TensorRef<Tensor<float, 3>> ref_of_ref(ref); in test_ref_of_ref()
126 TensorRef<Tensor<float, 3>> ref_of_ref2; in test_ref_of_ref()
154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorRef.h119 template<typename PlainObjectType> class TensorRef : public TensorBase<TensorRef<PlainObjectType> >
122 typedef TensorRef<PlainObjectType> Self;
144 EIGEN_STRONG_INLINE TensorRef() : m_evaluator(NULL) { in TensorRef() function
148 …EIGEN_STRONG_INLINE TensorRef(const Expression& expr) : m_evaluator(new internal::TensorLazyEvalua… in TensorRef() function
153 EIGEN_STRONG_INLINE TensorRef& operator = (const Expression& expr) {
160 ~TensorRef() { in ~TensorRef()
164 TensorRef(const TensorRef& other) : m_evaluator(other.m_evaluator) { in TensorRef() function
169 TensorRef& operator = (const TensorRef& other) {
356 struct TensorEvaluator<const TensorRef<Derived>, Device>
367 Layout = TensorRef<Derived>::Layout,
[all …]
DTensorTraits.h105 struct traits<TensorRef<PlainObjectType> >
158 struct eval<TensorRef<PlainObjectType>, Eigen::Dense>
160 typedef const TensorRef<PlainObjectType>& type;
164 struct eval<const TensorRef<PlainObjectType>, Eigen::Dense>
166 typedef const TensorRef<PlainObjectType>& type;
213 struct nested<TensorRef<PlainObjectType> >
215 typedef const TensorRef<PlainObjectType>& type;
219 struct nested<const TensorRef<PlainObjectType> >
221 typedef const TensorRef<PlainObjectType>& type;

12