| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast_test.cc | 45 EXPECT_EQ(4, bcast.output_batch_size()); in TEST()
58 EXPECT_EQ(1, bcast.output_batch_size()); in TEST()
71 EXPECT_EQ(48, bcast.output_batch_size()); in TEST()
84 EXPECT_EQ(6, bcast1.output_batch_size()); in TEST()
93 EXPECT_EQ(2, bcast2.output_batch_size()); in TEST()
123 EXPECT_EQ(12, bcast.output_batch_size()); in TEST()
135 EXPECT_EQ(12, bcast.output_batch_size()); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | arithmetic_builder.cc | 53 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
55 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
66 {output_batch_size, output_height_size, in PopulateSubGraph()
80 {output_batch_size, output_height_size, in PopulateSubGraph()
86 {output_batch_size, output_height_size, in PopulateSubGraph()
100 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | pool_2d_builder.cc | 61 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
63 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
68 {output_batch_size, output_height_size, in PopulateSubGraph()
76 {output_batch_size, output_height_size, in PopulateSubGraph()
98 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | reduce_builder.cc | 53 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
55 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
73 {output_batch_size, output_height_size, in PopulateSubGraph()
88 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | l2_normalization_builder.cc | 38 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
40 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
43 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | neg_op_builder.cc | 32 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
34 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
37 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | transpose_builder.cc | 37 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
39 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
42 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | squared_difference.cc | 59 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
61 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
65 {output_batch_size, output_height_size, in PopulateSubGraph()
82 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | hardswish_builder.cc | 40 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
42 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
46 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | quantize_builder.cc | 33 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
35 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
44 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | space_to_depth_builder.cc | 48 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
50 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
53 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | activation_builder.cc | 46 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
48 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
51 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | pad_builder.cc | 53 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
55 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
58 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | min_max_builder.cc | 43 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
45 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
48 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | softmax_builder.cc | 47 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
49 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
52 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | pack_builder.cc | 81 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
83 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
87 {output_batch_size, output_height_size, in PopulateSubGraph()
112 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | resize_nearest_neighbor_builder.cc | 67 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
69 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
72 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | arg_min_max_builder.cc | 60 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
65 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
68 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | split_builder.cc | 60 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
62 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
65 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | concat_builder.cc | 81 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
83 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
90 {output_batch_size, output_height_size, in PopulateSubGraph()
119 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | mirror_pad_builder.cc | 71 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
73 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
76 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | resize_bilinear_builder.cc | 67 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
69 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
72 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| D | reshape_builder.cc | 94 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
96 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
99 {output_batch_size, output_height_size, in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd.cc | 78 int output_batch_size = input_size->data[0]; in ResizeOutputTensor() local
81 TF_LITE_ENSURE_EQ(context, output_batch_size % block_shape[dim], 0); in ResizeOutputTensor()
82 output_batch_size = output_batch_size / block_shape[dim]; in ResizeOutputTensor()
87 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 60 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local
74 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND()
75 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND()
|