Searched refs:m_inputDepth (Results 1 – 4 of 4) sorted by relevance
64 m_inputDepth = input_dims[0]; in CustomTensorEvaluator()69 m_inputDepth = input_dims[NumInputDims - 1]; in CustomTensorEvaluator()207 m_planeInputStride = m_inputDepth; in CustomTensorEvaluator()208 m_rowInputStride = m_inputDepth * m_inputPlanes; in CustomTensorEvaluator()209 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in CustomTensorEvaluator()211 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in CustomTensorEvaluator()590 Index m_inputDepth; member
154 m_inputDepth = tensor.impl().impl().dimensions()[0]; in TensorContractionInputMapper()160 m_inputDepth = tensor.impl().impl().dimensions()[NumDims - 1]; in TensorContractionInputMapper()167 m_planeInputStride = m_inputDepth; in TensorContractionInputMapper()168 m_rowInputStride = m_inputDepth * m_inputPlanes; in TensorContractionInputMapper()169 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in TensorContractionInputMapper()171 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in TensorContractionInputMapper()230 m_inputDepth = base_mapper.m_inputDepth; in TensorContractionInputMapper()822 Index m_inputDepth; // Depth of the input tensor variable
185 m_inputDepth = input_dims[0];189 m_inputDepth = input_dims[NumInputDims-1];292 m_rowInputStride = m_inputDepth;293 m_colInputStride = m_inputDepth * m_inputRows;294 m_patchInputStride = m_inputDepth * m_inputRows * m_inputCols;488 Index m_inputDepth;
198 m_inputDepth = input_dims[0];203 m_inputDepth = input_dims[NumInputDims-1];316 m_planeInputStride = m_inputDepth;317 m_rowInputStride = m_inputDepth * m_inputPlanes;318 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes;319 m_otherInputStride = m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes;560 Index m_inputDepth;