Searched defs:Dims (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | argmax_op.h | 30 #define DECLARE_COMPUTE_SPEC(Dims) \ argument
50 #define DECLARE_COMPUTE_SPEC(Dims) \ argument
|
| D | in_topk_op.h | 45 using Dims = Eigen::DSizes<Eigen::Index, ndims>; member
|
| D | reverse_sequence_op.cc | 184 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ argument
193 #define DECLARE_GPU_SPEC_LEN(T, Dims) \ argument
|
| D | where_op.cc | 232 #define DECLARE_GPU_WHERE_INDEX(Dims, T, Tindex) \ argument
239 #define DECLARE_GPU_WHERE(Dims, T) \ argument
|
| D | in_topk_op_gpu.cu.cc | 89 using Dims = Eigen::DSizes<Eigen::Index, ndims>; typedef
|
| D | argmax_op.cc | 164 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
|
| D | pad_op.cc | 295 #define DECLARE_GPU_SPEC(T, Dims) \ argument
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | types.h | 26 struct Dims { struct
31 inline int Offset(const Dims<4>& dims, int i0, int i1, int i2, int i3) { in Offset() argument
|
| /external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AMDGPU/ |
| D | AMDGPUHSAMetadataStreamer.cpp | 194 std::vector<uint32_t> Dims; in getWorkGroupDimensions() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 124 struct Dims { struct
125 int sizes[N];
126 int strides[N];
189 inline int32 Dims(int i) const { in Dims() function
|
| /external/pdfium/third_party/lcms/src/ |
| D | cmslut.c | 467 cmsUInt32Number CubeSize(const cmsUInt32Number Dims[], cmsUInt32Number b) in CubeSize()
|