Searched refs:out_batch (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 74 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
89 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 378 int out_batch = GetTensorDim(input_shape, data_format_, 'N'); variable
456 ? ShapeFromFormat(data_format_, out_batch, out_rows, out_cols,
458 : ShapeFromFormat(data_format_, out_batch,
465 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
472 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = out_batch;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 469 const int64 out_batch = GetTensorDim(*output, params.data_format, 'N');
598 output_desc.set_count(out_batch)
622 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
|
| D | conv_ops.cc | 758 const int64 out_batch = GetTensorDim(*output, data_format, 'N'); in operator ()() local
886 output_desc.set_count(out_batch) in operator ()()
944 ShapeFromFormat(compute_data_format, out_batch, in operator ()()
|
| /external/kernel-headers/original/uapi/linux/ |
| D | bpf.h | 528 __aligned_u64 out_batch; /* output: next start batch */ member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5029 const int out_batch = in_batch % output_batch_size; in BatchToSpaceND() local
5057 T* out = output_data + Offset(output_shape, out_batch, out_h, out_w, 0); in BatchToSpaceND()
|