/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 | dynamic_slice_ops.cc | 49 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()
|
D | select_op.cc | 36 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()
|
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() 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 …]
|
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()
|
D | concat_op.cc | 47 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()
|
D | stateless_random_ops_v2.cc | 172 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 …]
|
D | listdiff_op.cc | 41 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()
|
D | sequence_ops.cc | 74 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()
|
D | pooling_ops.cc | 81 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 …]
|
D | clip_by_value_op.cc | 29 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()
|
D | reverse_op.cc | 37 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()
|
D | beta_op.cc | 37 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()
|
D | split_op.cc | 38 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()
|
D | sparse_to_dense_op.cc | 31 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/ |
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 | 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}) 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 …]
|
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 | 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 | 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 | 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() 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()
|
D | prelu_tester.h | 34 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/ |
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()
|