Home
last modified time | relevance | path

Searched defs:output_batch_size (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_builder.cc135 int output_batch_size, output_height_size, output_width_size, in BuildDilatedDwConv() local
251 int output_batch_size, output_height_size, output_width_size, in BuildSplittedDwConv() local
337 int output_batch_size, output_height_size, output_width_size, in BuildStandardConv() local
431 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dneg_op_builder.cc32 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dtranspose_builder.cc37 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dl2_normalization_builder.cc38 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dhardswish_builder.cc40 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dquantize_builder.cc33 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dspace_to_depth_builder.cc48 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dactivation_builder.cc46 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dpad_builder.cc53 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dsoftmax_builder.cc47 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dmin_max_builder.cc43 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Darg_min_max_builder.cc60 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dmirror_pad_builder.cc71 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dsplit_builder.cc59 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dresize_nearest_neighbor_builder.cc67 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dpool_2d_builder.cc61 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dresize_bilinear_builder.cc67 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Darithmetic_builder.cc53 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dstrided_slice_builder.cc73 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dsquared_difference.cc59 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dreshape_builder.cc94 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
H A Dslice_builder.cc70 int output_batch_size, output_height_size, output_width_size, in PopulateSubGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dbcast.h34 inline void ComputeBatchIndices(const int64_t output_batch_size, in ComputeBatchIndices()
99 const int64_t output_batch_size() const { return output_batch_size_; } in output_batch_size() function
H A Dmatmul_bcast.h69 const int64_t output_batch_size() const { return output_batch_size_; } in output_batch_size() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_to_space_nd.h60 const int output_batch_size = output_shape.Dims(0); in BatchToSpaceND() local

12