Searched defs:batch_size (Results 1 – 5 of 5) sorted by relevance
65 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local99 const int batch_size = SizeOfDimension(input_, 0); in EvalFloat32() local
66 const uint32_t batch_size = SizeOfDimension(input, 0); in Prepare() local154 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
57 uint32_t batch_size = getSizeOfDimension(outputShape, 0); in fullyConnectedFloat32() local216 uint32_t batch_size = input_n_elements / input_size; in prepare() local
39 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel()