Searched defs:batch_size (Results 1 – 7 of 7) sorted by relevance
72 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local106 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
83 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local170 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
65 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local149 const uint32_t batch_size = inputShape.dimensions[0]; in RNNStep() local
65 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32() local204 uint32_t batch_size = input_size == 0 ? 0 : input_n_elements / input_size; in validateShapes() local
39 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel()
22 batch_size = 2 variable