Home
last modified time | relevance | path

Searched refs:output_batch_size (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast_test.cc45 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()
119 EXPECT_EQ(12, bcast.output_batch_size()); in TEST()
131 EXPECT_EQ(12, bcast.output_batch_size()); in TEST()
Dbcast.h34 inline void ComputeBatchIndices(const int64 output_batch_size, in ComputeBatchIndices() argument
43 out_indices->resize(output_batch_size); in ComputeBatchIndices()
99 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size() function
Dmatmul_bcast.h39 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size() function
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/
Dreduce_builder.cc66 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
68 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
74 {output_batch_size, output_height_size, in PopulateSubGraph()
86 {output_batch_size, output_height_size, in PopulateSubGraph()
92 {output_batch_size, output_height_size, in PopulateSubGraph()
Dpool_2d_builder.cc70 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
72 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
77 {output_batch_size, output_height_size, in PopulateSubGraph()
85 {output_batch_size, output_height_size, in PopulateSubGraph()
109 {output_batch_size, output_height_size, in PopulateSubGraph()
Dmatmul_builder.cc107 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
109 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
112 {output_batch_size, output_height_size, in PopulateSubGraph()
125 {output_batch_size, output_height_size, in PopulateSubGraph()
156 {output_batch_size, output_height_size, in PopulateSubGraph()
187 {output_batch_size, output_height_size, in PopulateSubGraph()
Dconcat_builder.cc75 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
77 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
84 {output_batch_size, output_height_size, in PopulateSubGraph()
108 {output_batch_size, output_height_size, in PopulateSubGraph()
Dneg_op_builder.cc43 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()
Dl2_normalization_builder.cc52 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
54 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
57 {output_batch_size, output_height_size, in PopulateSubGraph()
Dactivation_builder.cc56 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
58 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
61 {output_batch_size, output_height_size, in PopulateSubGraph()
Dspace_to_depth_builder.cc63 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()
Dsoftmax_builder.cc58 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
60 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
63 {output_batch_size, output_height_size, in PopulateSubGraph()
Dtranspose_builder.cc60 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()
Dpad_builder.cc67 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()
Dresize_nearest_neighbor_builder.cc76 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
78 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
81 {output_batch_size, output_height_size, in PopulateSubGraph()
Darg_min_max_builder.cc71 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()
Dreshape_builder.cc94 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()
Dsplit_builder.cc73 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
75 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
78 {output_batch_size, output_height_size, in PopulateSubGraph()
Dresize_bilinear_builder.cc75 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
77 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
80 {output_batch_size, output_height_size, in PopulateSubGraph()
Darithmetic_builder.cc97 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
99 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
102 {output_batch_size, output_height_size, in PopulateSubGraph()
Dconv_2d_builder.cc213 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
215 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
310 {output_batch_size * dilation_factors_h_w_[0] * in PopulateSubGraph()
326 {output_batch_size, output_height_size, in PopulateSubGraph()
353 {output_batch_size, output_height_size, in PopulateSubGraph()
Dtranspose_conv_2d_builder.cc82 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
84 GetDims(&output_batch_size, &output_height_size, &output_width_size, in PopulateSubGraph()
143 {output_batch_size, output_height_size, in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd.cc82 const int output_batch_size = in ResizeOutputTensor() local
86 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
Dbatch_to_space_nd.cc80 const int output_batch_size = in ResizeOutputTensor() local
95 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_triangular_solve_op_impl.h124 const int64 batch_size = bcast.output_batch_size();
174 auto batch_size = bcast.output_batch_size();
285 for (int64 i = 0; i < bcast.output_batch_size(); ++i) {
347 const int64 batch_size = bcast.output_batch_size();

12