Home
last modified time | relevance | path

Searched refs:InputShape (Results 1 – 25 of 110) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc39 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 …]
Ddynamic_slice_ops.cc49 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
50 const TensorShape update_shape = ctx->InputShape("update"); in Compile()
51 const TensorShape index_shape = ctx->InputShape("indices"); in Compile()
86 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
87 const TensorShape start_indices_shape = ctx->InputShape("start_indices"); in Compile()
88 const TensorShape size_indices_shape = ctx->InputShape("size_indices"); in Compile()
Dselect_op.cc36 const TensorShape cond_shape = ctx->InputShape(0); in Compile()
37 const TensorShape then_shape = ctx->InputShape(1); in Compile()
38 const TensorShape else_shape = ctx->InputShape(2); in Compile()
87 const TensorShape cond_shape = ctx->InputShape(0); in Compile()
88 const TensorShape then_shape = ctx->InputShape(1); in Compile()
89 const TensorShape else_shape = ctx->InputShape(2); in Compile()
Dshape_op.cc45 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()
153 const TensorShape input_shape = ctx->InputShape(i); in Compile()
190 const TensorShape input_shape = ctx->InputShape(0); in Compile()
206 const TensorShape input_shape = ctx->InputShape(0); in Compile()
233 const TensorShape input_shape = ctx->InputShape("input"); in Compile()
234 const TensorShape dim_shape = ctx->InputShape("dim"); in Compile()
[all …]
Done_hot_op.cc33 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()
Dconcat_op.cc47 const TensorShape concat_dim_tensor_shape = ctx->InputShape(axis_index_); in Compile()
126 const TensorShape concat_dim_shape = ctx->InputShape(0); in Compile()
132 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(ctx->InputShape(i)), in Compile()
135 ctx->InputShape(i).DebugString())); in Compile()
156 const TensorShape inp0_shape = ctx->InputShape(1); in Compile()
171 const TensorShape inp_shape = ctx->InputShape(1 + i); in Compile()
Dstateless_random_ops_v2.cc172 auto alg_shape = ctx->InputShape(alg_input_idx); in GetAlgorithm()
217 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile()
219 ctx->InputShape(key_input_idx), in Compile()
270 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile()
272 ctx->InputShape(key_input_idx), in Compile()
277 TensorShape minval_shape = ctx->InputShape(minval_input_idx); in Compile()
281 TensorShape maxval_shape = ctx->InputShape(maxval_input_idx); in Compile()
331 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile()
333 ctx->InputShape(key_input_idx), in Compile()
377 auto counter_shape = ctx->InputShape(counter_input_idx); in Compile()
[all …]
Dlistdiff_op.cc41 OP_REQUIRES(context, TensorShapeUtils::IsVector(context->InputShape(0)), in Compile()
43 context->InputShape(0).DebugString())); in Compile()
45 OP_REQUIRES(context, TensorShapeUtils::IsVector(context->InputShape(1)), in Compile()
47 context->InputShape(1).DebugString())); in Compile()
Dsequence_ops.cc74 const TensorShape start_in_shape = ctx->InputShape(0); in Compile()
75 const TensorShape limit_in_shape = ctx->InputShape(1); in Compile()
76 const TensorShape delta_in_shape = ctx->InputShape(2); in Compile()
143 const TensorShape start_in_shape = ctx->InputShape("start"); in Compile()
144 const TensorShape stop_in_shape = ctx->InputShape("stop"); in Compile()
145 const TensorShape num_in_shape = ctx->InputShape("num"); in Compile()
Dpooling_ops.cc81 const TensorShape ksize_shape = ctx->InputShape(1); in GetKernelSize()
105 const TensorShape stride_shape = ctx->InputShape(2); in GetStride()
179 const TensorShape input_shape = ctx->InputShape(0); in Compile()
231 const TensorShape input_shape = ctx->InputShape(0); in Compile()
291 const TensorShape ksize_shape = ctx->InputShape(3); in Compile()
298 const TensorShape stride_shape = ctx->InputShape(4); in Compile()
315 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile()
316 const TensorShape tensor_out_shape = ctx->InputShape(1); in Compile()
317 const TensorShape out_backprop_shape = ctx->InputShape(2); in Compile()
428 const TensorShape out_backprop_shape = ctx->InputShape(1); in Compile()
[all …]
Dclip_by_value_op.cc29 const TensorShape shape = ctx->InputShape(0); in Compile()
30 const TensorShape min_shape = ctx->InputShape(1); in Compile()
31 const TensorShape max_shape = ctx->InputShape(2); in Compile()
Dreverse_op.cc37 const TensorShape x_shape = ctx->InputShape(0); in Compile()
38 const TensorShape revd_shape = ctx->InputShape(1); in Compile()
75 const TensorShape x_shape = ctx->InputShape(0); in Compile()
76 const TensorShape axes_shape = ctx->InputShape(1); in Compile()
Dbeta_op.cc37 const TensorShape& a_shape = ctx->InputShape(0); in Compile()
38 const TensorShape& b_shape = ctx->InputShape(1); in Compile()
39 const TensorShape& x_shape = ctx->InputShape(2); in Compile()
Dsplit_op.cc38 const TensorShape split_dim_shape = ctx->InputShape("split_dim"); in Compile()
39 const TensorShape input_shape = ctx->InputShape(1); in Compile()
105 const TensorShape input_shape = ctx->InputShape(0); in Compile()
106 const TensorShape index_shape = ctx->InputShape(2); in Compile()
135 const TensorShape split_size_shape = ctx->InputShape(1); in Compile()
Dsparse_to_dense_op.cc31 const TensorShape indices_shape = context->InputShape(0); in Compile()
51 const TensorShape sparse_values_shape = context->InputShape(2); in Compile()
62 const TensorShape default_value_shape = context->InputShape(3); in Compile()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dprelu_test.cc44 .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 …]
Dmean_test.cc43 .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}) in TEST()
148 .InputShape({batch, height, width, channels}) in TEST()
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 …]
Dreduce_tester.h36 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()
Dpad_test.cc43 .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 …]
Dfully_connected_test.cc41 .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 …]
Dreshape_test.cc44 .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()
Dpad_tester.cc39 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()
55 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()
Dprelu_tester.h34 inline PreluTester& InputShape(std::initializer_list<int32_t> shape) { in InputShape() function
42 inline const std::vector<int32_t>& InputShape() const { return input_shape_; } in InputShape() function
55 return InputShape(); in OutputShape()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dimage_resize_ops.cc44 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/
Dinplace_ops.cc45 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()

12345