Home
last modified time | relevance | path

Searched refs:input_rows_raw (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h130 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); in GetInputSizeInMklOrder() local
131 CHECK_BOUNDS(input_rows_raw, "Input rows too large"); in GetInputSizeInMklOrder()
132 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder()
154 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, '1'); in GetInputSizeInMklOrder() local
155 CHECK_BOUNDS(input_rows_raw, "Input rows too large"); in GetInputSizeInMklOrder()
156 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder()
Dconv_ops_using_gemm.cc490 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
493 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in Compute()
495 const int input_rows = static_cast<int>(input_rows_raw); in Compute()
Ddepthwise_conv_op.cc329 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute() local
332 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), in Compute()
334 const int32 input_rows = static_cast<int32>(input_rows_raw); in Compute()
Dconv_ops.cc371 const int64 input_rows_raw = GetTensorDim(input, params.data_format, 'H'); in ComputeConv2DDimension() local
372 TF_REQUIRES(FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
374 const int input_rows = static_cast<int>(input_rows_raw); in ComputeConv2DDimension()
Ddepthwise_conv_grad_op.cc70 const int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); \
73 FastBoundsCheck(input_rows_raw, std::numeric_limits<int32>::max()), \
75 const int32 input_rows = static_cast<int32>(input_rows_raw); \
Dmkl_conv_ops.cc488 const int64 input_rows_raw = in Compute() local
493 FastBoundsCheck(input_rows_raw, std::numeric_limits<int>::max()), in Compute()
495 const int input_rows = static_cast<int>(input_rows_raw); in Compute()