/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() 119 EXPECT_EQ(12, bcast.output_batch_size()); in TEST() 131 EXPECT_EQ(12, bcast.output_batch_size()); in TEST()
|
D | bcast.h | 34 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
|
D | matmul_bcast.h | 39 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size() function
|
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
D | reduce_builder.cc | 66 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()
|
D | pool_2d_builder.cc | 70 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()
|
D | matmul_builder.cc | 107 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()
|
D | concat_builder.cc | 75 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()
|
D | neg_op_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 | l2_normalization_builder.cc | 52 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()
|
D | activation_builder.cc | 56 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()
|
D | space_to_depth_builder.cc | 63 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 | softmax_builder.cc | 58 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()
|
D | transpose_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 | pad_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 | resize_nearest_neighbor_builder.cc | 76 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()
|
D | arg_min_max_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 | 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()
|
D | split_builder.cc | 73 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()
|
D | resize_bilinear_builder.cc | 75 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()
|
D | arithmetic_builder.cc | 97 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()
|
D | conv_2d_builder.cc | 213 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()
|
D | transpose_conv_2d_builder.cc | 82 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/ |
D | space_to_batch_nd.cc | 82 const int output_batch_size = in ResizeOutputTensor() local 86 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
|
D | batch_to_space_nd.cc | 80 const int output_batch_size = in ResizeOutputTensor() local 95 output_size->data[0] = output_batch_size; in ResizeOutputTensor()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | matrix_triangular_solve_op_impl.h | 124 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();
|