Searched refs:input_batch_raw (Results 1 – 1 of 1) sorted by relevance
130 int64 input_batch_raw = GetTensorDim(input_shape, data_format_, 'N'); in GetInputSizeInMklOrder() local131 CHECK_BOUNDS(input_batch_raw, "Input batch too large"); in GetInputSizeInMklOrder()132 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder()