/external/tensorflow/tensorflow/core/kernels/ |
depthwise_conv_op.cc | 322 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); variable 325 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), 327 const int32 input_rows = static_cast<int32>(input_rows_raw);
|
conv_ops_using_gemm.cc | 490 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); variable 493 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), 495 const int input_rows = static_cast<int>(input_rows_raw);
|
conv_ops.cc | 339 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); variable 342 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), 344 const int input_rows = static_cast<int>(input_rows_raw); [all...] |
mkl_conv_ops.h | 101 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); local 102 CHECK_BOUNDS(input_rows_raw, "Input rows too large"); 103 int input_rows = static_cast<int>(input_rows_raw);
|
mkl_conv_ops.cc | 137 const int64 input_rows_raw = variable 142 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), 144 const int input_rows = static_cast<int>(input_rows_raw); [all...] |