Searched defs:Dims (Results 1 – 10 of 10) 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 | reverse_sequence_op.cc | 189 #define DECLARE_GPU_SPEC(T, Tlen, Dims) \ argument
198 #define DECLARE_GPU_SPEC_LEN(T, Dims) \ argument
|
| D | in_topk_op.h | 45 using Dims = Eigen::DSizes<Eigen::Index, ndims>; member
|
| D | where_op.cc | 227 #define DECLARE_GPU_WHERE_INDEX(Dims, T, Tindex) \ argument
234 #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 | 177 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
|
| D | pad_op.cc | 294 #define DECLARE_GPU_SPEC(T, Dims) \ argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | runtime_shape.h | 28 struct Dims { struct
29 int sizes[N];
30 int strides[N];
86 inline int32_t Dims(int i) const { in Dims() function
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/ |
| D | AMDGPUHSAMetadataStreamer.cpp | 201 std::vector<uint32_t> Dims; in getWorkGroupDimensions() local
647 auto Dims = HSAMetadataDoc->getArrayNode(); in getWorkGroupDimensions() local
|
| /external/pdfium/third_party/lcms/src/ |
| D | cmslut.c | 461 cmsUInt32Number CubeSize(const cmsUInt32Number Dims[], cmsUInt32Number b) in CubeSize()
|