Searched refs:out_batch (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 372 int out_batch = GetTensorDim(input_shape, data_format_, 'N'); variable
447 ? ShapeFromFormat(data_format_, out_batch, out_rows, out_cols,
449 : ShapeFromFormat(data_format_, out_batch,
456 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
463 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = out_batch;
|
| D | fractional_avg_pool_op.cc | 244 const int64 out_batch = out_backprop.dim_size(0); in Compute() local
277 out_cols * out_rows * out_batch); in Compute()
282 for (int64 b = 0; b < out_batch; ++b) { in Compute()
|
| D | conv_ops_fused_impl.h | 592 const int64 out_batch = GetTensorDim(*output, params.data_format, 'N');
704 output_desc.set_count(out_batch)
728 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
|
| D | conv_ops.cc | 667 const int64 out_batch = GetTensorDim(*output, data_format, 'N'); in operator ()() local
770 output_desc.set_count(out_batch) in operator ()()
805 ShapeFromFormat(FORMAT_NCHW, out_batch, in operator ()()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3162 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
3177 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5808 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
5836 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6136 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
6164 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
|