Home
last modified time | relevance | path

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

12

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

12