| /external/tensorflow/tensorflow/core/kernels/ |
| fractional_avg_pool_op.cc | 244 const int64 out_batch = out_backprop.dim_size(0); variable
277 out_cols * out_rows * out_batch);
282 for (int64 b = 0; b < out_batch; ++b) {
|
| mkl_conv_ops.h | 237 int out_batch = GetTensorDim(input_shape, data_format_, 'N'); local
253 ShapeFromFormat(data_format_, out_batch, out_rows, out_cols, out_depth);
258 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
|
| conv_ops.cc | 564 const int64 out_batch = GetTensorDim(*output, data_format, 'N'); local
638 output_desc.set_count(out_batch)
670 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | [all...] |
