Searched defs:batch_size (Results 1 – 7 of 7) sorted by relevance
74 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local108 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
81 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local168 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
61 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local145 const uint32_t batch_size = inputShape.dimensions[0]; in RNNStep() local
66 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32() local206 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