Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h122 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable
425 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.h115 const size_t NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
143 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
Deigen_cuboid_convolution.h113 const int NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local
159 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
Deigen_volume_patch.h33 static constexpr 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/core/common_runtime/eager/
Dtensor_handle_data.cc50 Status LocalTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::LocalTensorHandleData
Dcustom_device_test.cc72 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::eager::__anon8e8d227f0111::TestCustomDeviceTensorHandle
Dtensor_handle.cc86 Status TensorHandle::PackedTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::TensorHandle::PackedTensorHandleData
627 Status TensorHandle::NumDims(int* num_dims) const { in NumDims() function in tensorflow::TensorHandle
/external/llvm-project/clang/include/clang/AST/
DExprOpenMP.h155 unsigned NumDims = 0; /// Number of dimensions in the shaping expression. variable
162 explicit OMPArrayShapingExpr(EmptyShell Shell, unsigned NumDims) in OMPArrayShapingExpr()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle_data.cc133 Status RemoteTensorHandleData::NumDims(int* num_dims) const { in NumDims() function in tensorflow::RemoteTensorHandleData
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMap.h240 const std::size_t NumDims = sizeof...(otherIndices) + 2; in operator() local
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc201 int TensorInterface::NumDims() const { return tensor_.dims(); } in NumDims() function in tensorflow::TensorInterface
/external/llvm-project/polly/lib/Support/
DISLTools.cpp210 int NumDims = Set.dim(isl::dim::set); in shiftDim() local
231 int NumDims = Map.dim(Dim); in shiftDim() local
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc507 Status NumDims(int* num_dims) const override { in NumDims() function in tensorflow::__anon6f1bef310211::CAPICustomDeviceTensorHandle
/external/llvm-project/clang/lib/Serialization/
DASTReaderStmt.cpp955 unsigned NumDims = Record.readInt(); in VisitOMPArrayShapingExpr() local
/external/llvm-project/polly/lib/CodeGen/
DPPCGCodeGeneration.cpp2837 unsigned NumDims = Array->getNumberOfDimensions(); in getExtent() local
/external/llvm-project/polly/lib/Analysis/
DScopBuilder.cpp2429 unsigned NumDims = Array->getNumberOfDimensions(); in foldSizeConstantsToRight() local
/external/llvm-project/clang/lib/AST/
DExpr.cpp4711 unsigned NumDims) { in CreateEmpty()