Searched defs:input_batch_size (Results 1 – 10 of 10) sorted by relevance
97 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() local232 int input_batch_size, input_height_size, input_width_size, input_depth_size; in PopulateSubGraph() local
65 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local
64 const int input_batch_size = input1_shape.Dims(0); in SpaceToBatchND() local
139 int64_t input_batch_size = orig_input_batch_size; in BatchToSpaceOpCompute() local
137 int64_t input_batch_size = orig_input_tensor.dim_size(0); in SpaceToBatchOpCompute() local
57 const int64_t input_batch_size = in PartitionConvolutionWithBatchGroupCount() local
380 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local451 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
379 const int32_t input_batch_size = input_shape[0]; in SpaceToBatchNDRank4() local450 const int32_t input_batch_size = input_shape[0]; in BatchToSpaceNDRank4() local
4553 const int input_batch_size = input1_shape.Dims(0); in BatchToSpaceND() local