| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 39 TensorShape alpha_shape = ctx->InputShape(1); in Compile()
44 TensorShape delta_shape = ctx->InputShape(2); in Compile()
84 TensorShape alpha_shape = ctx->InputShape(1); in Compile()
88 TensorShape l1_shape = ctx->InputShape(2); in Compile()
92 TensorShape l2_shape = ctx->InputShape(3); in Compile()
96 TensorShape delta_shape = ctx->InputShape(4); in Compile()
136 TensorShape lr_shape = ctx->InputShape(2); in Compile()
141 TensorShape grad_shape = ctx->InputShape(3); in Compile()
147 TensorShape momentum_shape = ctx->InputShape(4); in Compile()
194 TensorShape lr_shape = ctx->InputShape(2); in Compile()
[all …]
|
| D | select_op.cc | 37 const TensorShape cond_shape = ctx->InputShape(0); in Compile()
38 const TensorShape then_shape = ctx->InputShape(1); in Compile()
39 const TensorShape else_shape = ctx->InputShape(2); in Compile()
88 const TensorShape cond_shape = ctx->InputShape(0); in Compile()
89 const TensorShape then_shape = ctx->InputShape(1); in Compile()
90 const TensorShape else_shape = ctx->InputShape(2); in Compile()
|
| D | shape_op.cc | 45 const TensorShape input_shape = ctx->InputShape(0); in Compile()
77 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
78 const TensorShape bound_shape = ctx->InputShape("bound"); in Compile()
119 const TensorShape dim_index_shape = ctx->InputShape("dim_index"); in Compile()
120 const TensorShape size_shape = ctx->InputShape("size"); in Compile()
151 const TensorShape dim_index_shape = ctx->InputShape("dim_index"); in Compile()
181 const TensorShape input_shape = ctx->InputShape(i); in Compile()
218 const TensorShape input_shape = ctx->InputShape(0); in Compile()
234 const TensorShape input_shape = ctx->InputShape(0); in Compile()
261 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
[all …]
|
| D | one_hot_op.cc | 33 const TensorShape indices_shape = ctx->InputShape(0); in Compile()
34 const TensorShape depth_shape = ctx->InputShape(1); in Compile()
35 const TensorShape on_value_shape = ctx->InputShape(2); in Compile()
36 const TensorShape off_value_shape = ctx->InputShape(3); in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | prelu_test.cc | 44 .InputShape({batch, height, width, channels}) in TEST()
64 .InputShape({batch, height, width, channels})
85 .InputShape({batch, height, width, channels}) in TEST()
106 .InputShape({batch, height, width, channels}) in TEST()
127 .InputShape({batch, height, width, channels}) in TEST()
148 .InputShape({batch, height, width, channels}) in TEST()
169 .InputShape({batch, height, width, channels}) in TEST()
190 .InputShape({batch, height, width, channels}) in TEST()
210 .InputShape({batch, height, width, channels})
231 .InputShape({batch, height, width, channels}) in TEST()
[all …]
|
| D | unsigned_quantized_mean_test.cc | 44 .InputShape({batch, height, width, channels}) in TEST()
66 .InputShape({batch, height, width, channels}) in TEST()
88 .InputShape({batch, height, width, channels}) in TEST()
110 .InputShape({batch, height, width, channels}) in TEST()
132 .InputShape({batch, height, width, channels})
154 .InputShape({batch, height, width, channels})
176 .InputShape({batch, height, width, channels})
183 .InputShape({batch, height, width, channels})
205 .InputShape({batch, height, width, channels})
212 .InputShape({batch, height, width, channels})
[all …]
|
| D | mean_test.cc | 43 .InputShape({batch, height, width, channels}) in TEST()
64 .InputShape({batch, height, width, channels}) in TEST()
85 .InputShape({batch, height, width, channels}) in TEST()
106 .InputShape({batch, height, width, channels}) in TEST()
127 .InputShape({batch, height, width, channels})
148 .InputShape({batch, height, width, channels})
169 .InputShape({batch, height, width, channels})
175 .InputShape({batch, height, width, channels})
196 .InputShape({batch, height, width, channels})
202 .InputShape({batch, height, width, channels})
[all …]
|
| D | signed_quantized_mean_test.cc | 43 .InputShape({batch, height, width, channels}) in TEST()
64 .InputShape({batch, height, width, channels}) in TEST()
85 .InputShape({batch, height, width, channels}) in TEST()
106 .InputShape({batch, height, width, channels}) in TEST()
127 .InputShape({batch, height, width, channels})
148 .InputShape({batch, height, width, channels})
169 .InputShape({batch, height, width, channels})
175 .InputShape({batch, height, width, channels})
196 .InputShape({batch, height, width, channels})
202 .InputShape({batch, height, width, channels})
[all …]
|
| D | fully_connected_test.cc | 41 .InputShape({input_channels})
60 .InputShape({input_channels})
83 .InputShape({batch, input_channels})
105 .InputShape({batch, input_channels})
129 .InputShape({batch, width, input_channels})
152 .InputShape({batch, width, input_channels})
175 .InputShape({batch, width, input_channels})
200 .InputShape({batch, height, width, input_channels})
224 .InputShape({batch, height, width, input_channels})
247 .InputShape({batch, input_channels}) in TEST()
[all …]
|
| D | reduce_tester.h | 36 inline ReduceTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
45 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
68 output_shape.reserve(InputShape().size()); in OutputShape()
70 for (int32_t i = 0; i < InputShape().size(); i++) { in OutputShape()
76 output_shape.push_back(InputShape()[i]); in OutputShape()
85 for (int32_t i = 0; i < InputShape().size(); i++) { in OutputSize()
87 output_size *= InputShape()[i]; in OutputSize()
|
| D | pad_test.cc | 43 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
62 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
81 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
100 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
119 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
138 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
157 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
176 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
195 .InputShape({shape_rng(), shape_rng()}) in TEST()
214 .InputShape({shape_rng(), shape_rng()}) in TEST()
[all …]
|
| D | signed_quantized_pad_test.cc | 43 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
62 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
81 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
100 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
119 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
138 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
157 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
176 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
195 .InputShape({shape_rng(), shape_rng()}) in TEST()
214 .InputShape({shape_rng(), shape_rng()}) in TEST()
[all …]
|
| D | quantized_reduce_tester.h | 37 inline QuantizedReduceTester& InputShape( in InputShape() function
47 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
70 output_shape.reserve(InputShape().size()); in OutputShape()
72 for (int32_t i = 0; i < InputShape().size(); i++) { in OutputShape()
78 output_shape.push_back(InputShape()[i]); in OutputShape()
87 for (int32_t i = 0; i < InputShape().size(); i++) { in OutputSize()
89 output_size *= InputShape()[i]; in OutputSize()
|
| D | unsigned_quantized_pad_test.cc | 44 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
64 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
84 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
104 .InputShape({shape_rng(), shape_rng(), shape_rng(), shape_rng()}) in TEST()
124 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
144 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
164 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
184 .InputShape({shape_rng(), shape_rng(), shape_rng()}) in TEST()
204 .InputShape({shape_rng(), shape_rng()}) in TEST()
224 .InputShape({shape_rng(), shape_rng()}) in TEST()
[all …]
|
| D | reshape_test.cc | 44 .InputShape(input_shape)
65 .InputShape(input_shape)
86 .InputShape(input_shape)
107 .InputShape(input_shape)
127 .InputShape(input_shape)
147 .InputShape(input_shape)
165 .InputShape(shape)
183 .InputShape(shape)
195 .InputShape(std::vector<int32_t>())
218 .InputShape(input_shape) in TEST()
|
| D | signed_quantized_split_test.cc | 43 .InputShape(shape)
69 .InputShape(shape)
95 .InputShape(shape)
122 .InputShape(shape)
144 .InputShape(shape)
170 .InputShape(shape)
196 .InputShape(shape)
223 .InputShape(shape)
245 .InputShape(shape)
271 .InputShape(shape)
[all …]
|
| D | split_test.cc | 43 .InputShape(shape)
69 .InputShape(shape)
95 .InputShape(shape)
122 .InputShape(shape)
144 .InputShape(shape)
170 .InputShape(shape)
196 .InputShape(shape)
223 .InputShape(shape)
245 .InputShape(shape)
271 .InputShape(shape)
[all …]
|
| D | unsigned_quantized_split_test.cc | 43 .InputShape(shape)
69 .InputShape(shape)
95 .InputShape(shape)
122 .InputShape(shape)
144 .InputShape(shape)
170 .InputShape(shape)
196 .InputShape(shape)
223 .InputShape(shape)
245 .InputShape(shape)
271 .InputShape(shape)
[all …]
|
| D | pad_tester.cc | 40 output_shape.reserve(InputShape().size()); in OutputShape()
41 for (size_t i = 0; i < InputShape().size(); i++) { in OutputShape()
42 int32_t output_dim = InputShape()[i]; in OutputShape()
56 ASSERT_LE(InputPrePaddings().size(), InputShape().size()); in Test()
97 default_input_data + ComputeSize(InputShape()), in Test()
102 std::copy(default_input_data, default_input_data + ComputeSize(InputShape()), in Test()
142 builder.CreateVector<int32_t>(InputShape().data(), in CreateTfLiteModel()
143 InputShape().size()), in CreateTfLiteModel()
|
| D | signed_quantized_fully_connected_test.cc | 47 .InputShape({input_channels})
72 .InputShape({input_channels})
101 .InputShape({batch, input_channels})
129 .InputShape({batch, input_channels})
159 .InputShape({batch, width, input_channels})
188 .InputShape({batch, width, input_channels})
217 .InputShape({batch, width, input_channels})
248 .InputShape({batch, height, width, input_channels})
278 .InputShape({batch, height, width, input_channels})
307 .InputShape({batch, input_channels}) in TEST()
[all …]
|
| D | quantized_pad_tester.cc | 39 output_shape.reserve(InputShape().size()); in OutputShape()
40 for (size_t i = 0; i < InputShape().size(); i++) { in OutputShape()
41 int32_t output_dim = InputShape()[i]; in OutputShape()
65 default_input_data + ComputeSize(InputShape()), in Test()
69 std::copy(default_input_data, default_input_data + ComputeSize(InputShape()), in Test()
85 ASSERT_LE(InputPrePaddings().size(), InputShape().size()); in Test()
150 builder.CreateVector<int32_t>(InputShape().data(), in CreateTfLiteModel()
151 InputShape().size()), in CreateTfLiteModel()
|
| D | split_tester.h | 41 inline SplitTester& InputShape(const std::vector<int32_t>& shape) { in InputShape() function
58 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
61 std::vector<int32_t> output_shape = InputShape(); in OutputShape()
63 split_dim += split_dim < 0 ? InputShape().size() : 0; in OutputShape()
|
| D | split_tester.cc | 49 default_input_data + ComputeSize(InputShape()), in Test()
53 std::copy(default_input_data, default_input_data + ComputeSize(InputShape()), in Test()
82 default_input_data + ComputeSize(InputShape()), in Test()
87 std::copy(default_input_data, default_input_data + ComputeSize(InputShape()), in Test()
113 axis += axis < 0 ? InputShape().size() : 0; in Test()
114 ASSERT_EQ(0, InputShape()[axis] % NumSplits()); in Test()
185 builder.CreateVector<int32_t>(InputShape().data(), in CreateTfLiteModel()
186 InputShape().size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 44 TensorShapeToXLAShape(ctx->output_xla_type(0), ctx->InputShape(0)); in GetOutputShape()
57 TensorShapeToXLAShape(ctx->output_xla_type(0), ctx->InputShape(0)); in CompileGrad()
58 if (ctx->InputShape(1).dim_sizes() == ctx->InputShape(0).dim_sizes()) { in CompileGrad()
78 if (ctx->InputShape(0).dim_size(1) == output_shape.dimensions(1) && in CompileForward()
79 ctx->InputShape(0).dim_size(2) == output_shape.dimensions(2)) { in CompileForward()
84 if (ctx->InputShape(0).dim_size(1) == 1 && in CompileForward()
85 ctx->InputShape(0).dim_size(2) == 1) { in CompileForward()
132 TensorShapeToXLAShape(ctx->output_xla_type(0), ctx->InputShape(1)); in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | inplace_ops.cc | 45 const TensorShape x_shape = ctx->InputShape(0); in Compile()
46 const TensorShape i_shape = ctx->InputShape(1); in Compile()
47 const TensorShape v_shape = ctx->InputShape(2); in Compile()
105 const TensorShape x_shape = ctx->InputShape(0); in Compile()
106 const TensorShape i_shape = ctx->InputShape(1); in Compile()
107 const TensorShape v_shape = ctx->InputShape(2); in Compile()
|