Searched refs:input_rows_raw (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 136 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); in GetInputSizeInMklOrder() local
137 CHECK_BOUNDS(input_rows_raw, "Input rows too large"); in GetInputSizeInMklOrder()
138 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder()
160 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, '1'); in GetInputSizeInMklOrder() local
161 CHECK_BOUNDS(input_rows_raw, "Input rows too large"); in GetInputSizeInMklOrder()
162 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_using_gemm.cc | 492 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
495 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in Compute()
497 const int input_rows = static_cast<int>(input_rows_raw); in Compute()
|
| D | depthwise_conv_op.cc | 355 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
358 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), in Compute()
360 const int32 input_rows = static_cast<int32>(input_rows_raw); in Compute()
|
| D | conv_ops.cc | 438 const int64 input_rows_raw = GetTensorDim(input, params.data_format, 'H'); in ComputeConv2DDimension() local
439 TF_REQUIRES(FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
441 const int input_rows = static_cast<int>(input_rows_raw); in ComputeConv2DDimension()
|
| D | depthwise_conv_grad_op.cc | 75 const int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); \
78 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), \
80 const int32 input_rows = static_cast<int32>(input_rows_raw); \
|