Home
last modified time | relevance | path

Searched defs:Dims (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dargmax_op.h30 #define DECLARE_COMPUTE_SPEC(Dims) \ argument
50 #define DECLARE_COMPUTE_SPEC(Dims) \ argument
Din_topk_op.h45 using Dims = Eigen::DSizes<Eigen::Index, ndims>; member
Dreverse_sequence_op.cc186 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ argument
195 #define DECLARE_GPU_SPEC_LEN(T, Dims) \ argument
Dwhere_op.cc227 #define DECLARE_GPU_WHERE_INDEX(Dims, T, Tindex) \ argument
234 #define DECLARE_GPU_WHERE(Dims, T) \ argument
Din_topk_op_gpu.cu.cc89 using Dims = Eigen::DSizes<Eigen::Index, ndims>; typedef
Dargmax_op.cc165 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
Dpad_op.cc292 #define DECLARE_GPU_SPEC(T, Dims) \ argument
/external/tensorflow/tensorflow/core/kernels/neon/
Dtypes.h26 struct Dims { struct
31 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) { in Offset() argument
/external/llvm-project/polly/lib/Transform/
DFlattenAlgo.cpp133 unsigned Dims = 0; in scheduleScatterDims() local
178 auto Dims = ScatterSet.dim(isl::dim::set); in tryFlattenSequence() local
309 auto Dims = scheduleScatterDims(Schedule); in flattenSchedule() local
DScheduleOptimizer.cpp307 unsigned Dims = IsolateDomain.dim(isl::dim::set); in getIsolateOptions() local
358 unsigned Dims = Set.dim(isl::dim::set); in addExtentConstraints() local
372 unsigned Dims = ScheduleRange.dim(isl::dim::set); in getPartialTilePrefixes() local
439 auto Dims = Space.dim(isl::dim::set); in tileNode() local
505 auto Dims = Space.dim(isl::dim::set); in isTileableBandNode() local
537 auto Dims = Space.dim(isl::dim::set); in standardBandOpts() local
1204 unsigned Dims = Prefix.dim(isl::dim::set); in isolateAndUnrollMatMulInnerLoops() local
/external/llvm-project/polly/lib/Support/
DISLTools.cpp161 unsigned Dims = 0; in getNumScatterDims() local
175 unsigned Dims = getNumScatterDims(Schedule); in getScatterSpace() local
772 int Dims = BSet.dim(isl::dim::set); in recursiveExpand() local
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/
DAMDGPUHSAMetadataStreamer.cpp201 std::vector<uint32_t> Dims; in getWorkGroupDimensions() local
647 auto Dims = HSAMetadataDoc->getArrayNode(); in getWorkGroupDimensions() local
/external/llvm-project/llvm/lib/Target/AMDGPU/
DAMDGPUHSAMetadataStreamer.cpp186 std::vector<uint32_t> Dims; in getWorkGroupDimensions() local
604 auto Dims = HSAMetadataDoc->getArrayNode(); in getWorkGroupDimensions() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h143 struct Dims { struct
144 int sizes[N];
145 int strides[N];
213 inline int32_t Dims(int i) const { in Dims() function
/external/pdfium/third_party/lcms/src/
Dcmslut.c467 cmsUInt32Number CubeSize(const cmsUInt32Number Dims[], cmsUInt32Number b) in CubeSize()
/external/llvm-project/clang/lib/AST/
DExpr.cpp4671 void OMPArrayShapingExpr::setDimensions(ArrayRef<Expr *> Dims) { in setDimensions()
4687 ArrayRef<Expr *> Dims) in OMPArrayShapingExpr()
4698 ArrayRef<Expr *> Dims, in Create()
/external/llvm-project/clang/lib/Serialization/
DASTReaderStmt.cpp957 SmallVector<Expr *, 4> Dims(NumDims); in VisitOMPArrayShapingExpr() local
/external/llvm-project/polly/lib/Analysis/
DScopInfo.cpp1105 auto Dims = SAI->getNumberOfDimensions(); in setNewAccessRelation() local
DScopBuilder.cpp2374 int Dims = Elements.dim(isl::dim::set); in foldSizeConstantsToRight() local
/external/llvm-project/clang/lib/Sema/
DTreeTransform.h2466 ArrayRef<Expr *> Dims, in RebuildOMPArrayShapingExpr()
10394 SmallVector<Expr *, 4> Dims; in TransformOMPArrayShapingExpr() local
DSemaExpr.cpp5040 ArrayRef<Expr *> Dims, in ActOnOMPArrayShapingExpr()
/external/llvm-project/clang/lib/CodeGen/
DCGOpenMPRuntime.cpp7107 MapDimArrayTy Dims; member