Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:kNumSpatialDims (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc	81 constexpr int kNumSpatialDims = 2; 83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); 85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); 87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); 89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) *
spacetodepth_op.cc	81 constexpr int kNumSpatialDims = 2; 83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); 85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); 87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); 89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) *

Completed in 96 milliseconds