Searched refs:input_cols_raw (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 135 int64 input_cols_raw = GetTensorDim(input_shape, data_format_, 'W'); in GetInputSizeInMklOrder() local
136 CHECK_BOUNDS(input_cols_raw, "Input cols too large"); in GetInputSizeInMklOrder()
137 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder()
159 int64 input_cols_raw = GetTensorDim(input_shape, data_format_, '2'); in GetInputSizeInMklOrder() local
160 CHECK_BOUNDS(input_cols_raw, "Input cols too large"); in GetInputSizeInMklOrder()
161 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder()
|
| D | conv_ops_using_gemm.cc | 500 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); in Compute() local
503 FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in Compute()
505 const int input_cols = static_cast<int>(input_cols_raw); in Compute()
|
| D | depthwise_conv_op.cc | 337 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); in Compute() local
340 FastBoundsCheck(input_cols_raw, std::numeric_limits<int32>::max()), in Compute()
342 const int32 input_cols = static_cast<int32>(input_cols_raw); in Compute()
|
| D | conv_ops.cc | 379 const int64 input_cols_raw = GetTensorDim(input, params.data_format, 'W'); in ComputeConv2DDimension() local
380 TF_REQUIRES(FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
382 const int input_cols = static_cast<int>(input_cols_raw); in ComputeConv2DDimension()
|
| D | depthwise_conv_grad_op.cc | 76 const int64 input_cols_raw = GetTensorDim(input_shape, data_format_, 'W'); \
79 FastBoundsCheck(input_cols_raw, std::numeric_limits<int32>::max()), \
81 const int32 input_cols = static_cast<int32>(input_cols_raw); \
|
| D | mkl_conv_ops.cc | 500 const int64 input_cols_raw = in Compute() local
505 FastBoundsCheck(input_cols_raw, std::numeric_limits<int>::max()), in Compute()
507 const int input_cols = static_cast<int>(input_cols_raw); in Compute()
|