Home
last modified time | relevance | path

Searched defs:NumDims (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h121 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
374 static const int NumDims = internal::traits<Input>::NumDimensions; variable
Deigen_backward_cuboid_convolutions.h112 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
412 static const int NumDims = internal::traits<Input>::NumDimensions; variable
Deigen_spatial_convolutions-inl.h111 const size_t NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
138 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
Deigen_cuboid_convolution.h111 const int NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
157 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
Deigen_volume_patch.h33 static const int NumDims = NumInputDims + 1; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h115 const int NumDims = internal::traits<Input>::NumDimensions; variable
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h118 const int NumDims = internal::traits<Input>::NumDimensions; variable
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc187 Status TensorHandle::NumDims(int* num_dims) { in NumDims() function in tensorflow::TensorHandle
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMap.h240 const std::size_t NumDims = sizeof...(otherIndices) + 2; in operator() local