HomeSort by relevance Sort by last modified time
    Searched refs:input_batch_raw (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h 111 int64 input_batch_raw = GetTensorDim(input_shape, data_format_, 'N'); local
112 CHECK_BOUNDS(input_batch_raw, "Input batch too large");
113 int input_batch = static_cast<int>(input_batch_raw);
mkl_conv_ops.cc 160 const int64 input_batch_raw = variable
165 FastBoundsCheck(input_batch_raw, std::numeric_limits<int>::max()),
167 const int batch = static_cast<int>(input_batch_raw);
    [all...]

Completed in 845 milliseconds