Searched refs:input_planes_raw (Results 1 – 1 of 1) sorted by relevance
155 int64 input_planes_raw = GetTensorDim(input_shape, data_format_, '0'); in GetInputSizeInMklOrder() local156 CHECK_BOUNDS(input_planes_raw, "Input depth too large"); in GetInputSizeInMklOrder()157 int input_planes = static_cast<int>(input_planes_raw); in GetInputSizeInMklOrder()