Home
last modified time | relevance | path

Searched defs:input_batch_size (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc97 int input_batch_size, int input_height_size, int input_width_size, in PopulateSpaceToBatchOutputDims()
128 int input_batch_size, input_height_size, input_width_size, input_depth_size; in PopulateSubGraph() local
232 int input_batch_size, input_height_size, input_width_size, input_depth_size; in PopulateSubGraph() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h65 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
Dspace_to_batch_nd.h64 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_op.cc137 int64 input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local
Dbatchtospace_op.cc139 int64 input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc57 const int64 input_batch_size = in PartitionConvolutionWithBatchGroupCount() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5019 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local