Searched refs:input_batch_raw (Results 1 – 2 of 2) sorted by relevance
124 int64 input_batch_raw = GetTensorDim(input_shape, data_format_, 'N'); in GetInputSizeInMklOrder() local125 CHECK_BOUNDS(input_batch_raw, "Input batch too large"); in GetInputSizeInMklOrder()126 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder()
511 const int64 input_batch_raw = in Compute() local516 FastBoundsCheck(input_batch_raw, std::numeric_limits<int>::max()), in Compute()518 const int batch = static_cast<int>(input_batch_raw); in Compute()