Searched refs:batch_raw (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_using_gemm.cc | 509 const int64 batch_raw = GetTensorDim(input, data_format_, 'N'); in Compute() local
511 FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in Compute()
513 const int batch = static_cast<int>(batch_raw); in Compute()
|
| D | conv_ops_fused_image_transform.cc | 814 const int64 batch_raw = padded_shape.dim_size(0); in Compute() local
816 FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in Compute()
818 const int batch = static_cast<int>(batch_raw); in Compute()
|
| D | conv_ops.cc | 386 const int64 batch_raw = GetTensorDim(input, params.data_format, 'N'); in ComputeConv2DDimension() local
387 TF_REQUIRES(FastBoundsCheck(batch_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
389 const int batch = static_cast<int>(batch_raw); in ComputeConv2DDimension()
|