Searched refs:TF_REQUIRES (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 275 #define TF_REQUIRES(EXP, STATUS) \ macro
291 TF_REQUIRES(FormatFromString(data_format_string, ¶ms->data_format), in InitConv2DParameters()
298 TF_REQUIRES(dilations.size() == 4, in InitConv2DParameters()
301 TF_REQUIRES(strides.size() == 4, in InitConv2DParameters()
308 TF_REQUIRES( in InitConv2DParameters()
312 TF_REQUIRES(stride_h > 0 && stride_w > 0, in InitConv2DParameters()
320 TF_REQUIRES( in InitConv2DParameters()
324 TF_REQUIRES( in InitConv2DParameters()
339 TF_REQUIRES(input.dims() == 4, in ComputeConv2DDimension()
342 TF_REQUIRES(filter.dims() == 4, in ComputeConv2DDimension()
[all …]
|
| D | conv_ops_fused_impl.h | 385 #define TF_REQUIRES(EXP, STATUS) \
396 TF_REQUIRES(bias.dims() == 1,
412 TF_REQUIRES(scale.dims() == 1,
415 TF_REQUIRES(offset.dims() == 1,
418 TF_REQUIRES(estimated_mean.dims() == 1,
421 TF_REQUIRES(
439 #undef TF_REQUIRES
|