Home
last modified time | relevance | path

Searched refs:DIM (Results 1 – 25 of 67) sorted by relevance

123

/external/llvm-project/mlir/test/Dialect/Shape/
Dshape-to-shape.mlir11 // CHECK: ^bb0({{.*}}: index, [[DIM:%.*]]: !shape.size, [[ACC:%.*]]: !shape.size
12 // CHECK: [[NEW_ACC:%.*]] = shape.mul [[DIM]], [[ACC]]
27 // CHECK: ^bb0({{.*}}: index, [[DIM:%.*]]: index, [[ACC:%.*]]: index
28 // CHECK: [[NEW_ACC:%.*]] = shape.mul [[DIM]], [[ACC]]
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h243 #define INSTANTIATE_DIM1_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
244 template void HandleStridedSliceCase<DEVICE, T, DIM>( \
250 template void HandleStridedSliceGradCase<DEVICE, T, DIM>( \
257 #define INSTANTIATE_DIM0_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
258 template class HandleStridedSliceAssignCase<DEVICE, T, DIM>;
262 #define INSTANTIATE(DEVICE, T, DIM) \ argument
263 INSTANTIATE_DIM0_AND_UP_HANDLERS(DEVICE, T, DIM)
265 #define INSTANTIATE(DEVICE, T, DIM) \ argument
266 INSTANTIATE_DIM0_AND_UP_HANDLERS(DEVICE, T, DIM) \
267 INSTANTIATE_DIM1_AND_UP_HANDLERS(DEVICE, T, DIM)
Dreverse_op_gpu.cu.cc28 #define DEFINE_REVERSE(T, DIM) \ argument
29 template struct functor::Reverse<GPUDevice, T, DIM>;
Dreverse_op.cc322 #define DECLARE_GPU_SPEC_DIM(T, DIM) \ argument
324 void Reverse<GPUDevice, T, DIM>::operator()( \
325 const GPUDevice& d, typename TTypes<T, DIM>::ConstTensor input, \
326 const Eigen::array<bool, DIM>& reverse_dims, \
327 typename TTypes<T, DIM>::Tensor output); \
328 extern template struct Reverse<GPUDevice, T, DIM>;
Dtranspose_functor_gpu.cu.cc172 #define HANDLE_DIM(DIM) \ argument
173 case DIM: \
174 internal::TransposeUsingEigen<GPUDevice, T, DIM>(d, in, perm, conjugate, \
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dmhlo_infer_shape_type_methods.mlir11 // CHECK: %[[DIM:.*]] = index_cast %[[DIM_AS_INDEX]] : index to i64
12 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[C2]], %[[DIM]] : tensor<2xi64>
28 // CHECK: %[[DIM:.*]] = index_cast %[[DIM_AS_INDEX]] : index to i64
29 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[C2]], %[[DIM]] : tensor<2xi64>
Dunfuse_batch_norm.mlir111 // CHECK-DAG: %[[DIM:.+]] = dim %[[VARIANCE]], %[[C0]] : tensor<?xf32>
112 // CHECK-DAG: %[[TO_DIM_TENSOR:.+]] = tensor.from_elements %[[DIM]] : tensor<1xindex>
/external/deqp/external/vulkancts/modules/vulkan/binding_model/
DvktBindingBufferDeviceAddressTests.cpp63 static const deUint32 DIM = 8; variable
502 …" imageStore(image0_0, ivec2(gl_VertexIndex % " << DIM << ", gl_VertexIndex / " << DIM << "), col… in initPrograms()
882 …vk, device, allocator, makeBufferCreateInfo(DE_NULL, DIM*DIM*sizeof(deUint32), VK_BUFFER_USAGE_TRA… in iterate()
892 DIM, // deUint32 width; in iterate()
893 DIM, // deUint32 height; in iterate()
1088 DIM, // width in iterate()
1089 DIM, // height in iterate()
1145 VkViewport viewport = makeViewport(DIM, DIM); in iterate()
1146 VkRect2D scissor = makeRect2D(DIM, DIM); in iterate()
1274 vk.cmdDispatch(*cmdBuffer, DIM, DIM, 1); in iterate()
[all …]
DvktBindingDescriptorSetRandomTests.cpp76 static const deUint32 DIM = 8; variable
911 const auto expectedInvocationID = descriptor % (DIM*DIM); in initPrograms()
964 …" const int invocationID = int(gl_GlobalInvocationID.y) * " << DIM << " + int(gl_GlobalInvocation… in initPrograms()
985 " const int invocationID = int(gl_LaunchIDNV.y) * " << DIM << " + int(gl_LaunchIDNV.x);\n" in initPrograms()
1006 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
1033 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
1065 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
1095 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
1124 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
1166 " const int invocationID = int(gl_LaunchIDEXT.y) * " << DIM << " + int(gl_LaunchIDEXT.x);\n" in initPrograms()
[all …]
/external/sl4a/Common/src/com/googlecode/android_scripting/facade/
DWakeLockFacade.java41 FULL, PARTIAL, BRIGHT, DIM enumConstant
53 addWakeLock(WakeLockType.DIM, PowerManager.SCREEN_DIM_WAKE_LOCK in WakeLockManager()
105 mManager.acquire(WakeLockType.DIM); in wakeLockAcquireDim()
/external/mesa3d/src/mesa/swrast/
Ds_texfetch_tmp.h44 #if DIM == 1
51 #elif DIM == 2
60 #elif DIM == 3
187 #undef DIM
Ds_texfetch.c54 #define DIM 1 macro
57 #define DIM 2 macro
60 #define DIM 3 macro
/external/deqp/external/vulkancts/modules/vulkan/memory_model/
DvktMemoryModelMessagePassing.cpp1093 deUint32 DIM = 31; in iterate() local
1096 if (DIM*DIM > properties.properties.limits.maxComputeWorkGroupInvocations) in iterate()
1098 DIM = (deUint32)deFloatSqrt((float)properties.properties.limits.maxComputeWorkGroupInvocations); in iterate()
1100 deUint32 NUM_INVOCATIONS = (DIM * DIM * NUM_WORKGROUP_EACH_DIM * NUM_WORKGROUP_EACH_DIM); in iterate()
1201 DIM*NUM_WORKGROUP_EACH_DIM, // deUint32 width; in iterate()
1202 DIM*NUM_WORKGROUP_EACH_DIM, // deUint32 height; in iterate()
1369 const deUint32 specData[2] = {DIM, NUM_WORKGROUP_EACH_DIM}; in iterate()
1452 DIM*NUM_WORKGROUP_EACH_DIM, // width in iterate()
1453 DIM*NUM_WORKGROUP_EACH_DIM, // height in iterate()
1509 VkViewport viewport = makeViewport(DIM*NUM_WORKGROUP_EACH_DIM, DIM*NUM_WORKGROUP_EACH_DIM); in iterate()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h913 template <int N, int DIM, typename Calc>
914 typename std::enable_if<DIM != N - 1, void>::type NDOpsHelperImpl( in NDOpsHelperImpl()
916 for (indexes[DIM] = 0; indexes[DIM] < output.extents[DIM]; ++indexes[DIM]) { in NDOpsHelperImpl()
917 NDOpsHelperImpl<N, DIM + 1, Calc>(output, calc, indexes); in NDOpsHelperImpl()
921 template <int N, int DIM, typename Calc>
922 typename std::enable_if<DIM == N - 1, void>::type NDOpsHelperImpl( in NDOpsHelperImpl()
924 for (indexes[DIM] = 0; indexes[DIM] < output.extents[DIM]; ++indexes[DIM]) { in NDOpsHelperImpl()
/external/skqp/tests/
DImageCacheTest.cpp43 static const int DIM = 256; variable
117 static const size_t defLimit = DIM * DIM * 4 * COUNT + 1024; // 1K slop in DEF_TEST()
/external/skia/tests/
DImageCacheTest.cpp43 static const int DIM = 256; variable
117 static const size_t defLimit = DIM * DIM * 4 * COUNT + 1024; // 1K slop in DEF_TEST()
/external/deqp/external/vulkancts/modules/vulkan/robustness/
DvktRobustnessExtsTests.cpp149 static const deUint32 DIM = 8; variable
1462 …" imageStore(image0_0, ivec2(gl_VertexIndex % " << DIM << ", gl_VertexIndex / " << DIM << "), col… in initPrograms()
1835 DIM, // deUint32 width; in iterate()
1836 DIM, // deUint32 height; in iterate()
2214 …vk, device, allocator, makeBufferCreateInfo(DIM*DIM*16, VK_BUFFER_USAGE_TRANSFER_DST_BIT), MemoryR… in iterate()
2520 DIM, // width in iterate()
2521 DIM, // height in iterate()
2594 VkViewport viewport = makeViewport(DIM, DIM); in iterate()
2595 VkRect2D scissor = makeRect2D(DIM, DIM); in iterate()
2736 vk.cmdDispatch(*cmdBuffer, DIM, DIM, 1); in iterate()
[all …]
/external/mesa3d/src/compiler/glsl/tests/
Dsampler_types_test.cpp40 #define T(TYPE, DIM, DATA_TYPE, ARR, SHAD, COMPS) \ argument
45 EXPECT_EQ(DIM, type->sampler_dimensionality); \
/external/llvm-project/mlir/test/Dialect/Affine/
Daffine-loop-normalize.mlir80 // CHECK-NEXT: %[[DIM:.*]] = dim %arg0, %c0 : memref<?x?xf32>
81 // CHECK-NEXT: affine.for %[[I:.*]] = 0 to [[$UB00]]()[%[[DIM]]] {
111 // CHECK-NEXT: %[[DIM:.*]] = dim %arg0, %c0 : memref<?x?xf32>
112 // CHECK-NEXT: affine.for %[[I:.*]] = 0 to [[$OUTERUB]]()[%[[DIM]]] {
/external/llvm-project/mlir/test/Conversion/VectorToSCF/
Dvector-to-loops.mlir214 // FULL-UNROLL: %[[DIM:.*]] = dim %[[A]], %[[C0]] : memref<?x?xf32>
215 // FULL-UNROLL: cmpi "slt", %[[base]], %[[DIM]] : index
225 // FULL-UNROLL: cmpi "slt", %{{.*}}, %[[DIM]] : index
235 // FULL-UNROLL: cmpi "slt", %{{.*}}, %[[DIM]] : index
281 // FULL-UNROLL: %[[DIM:.*]] = dim %[[A]], %[[C0]] : memref<?x?xf32>
282 // FULL-UNROLL: %[[CMP0:.*]] = cmpi "slt", %[[base]], %[[DIM]] : index
288 // FULL-UNROLL: %[[CMP1:.*]] = cmpi "slt", %[[I1]], %[[DIM]] : index
294 // FULL-UNROLL: %[[CMP2:.*]] = cmpi "slt", %[[I2]], %[[DIM]] : index
/external/llvm-project/mlir/test/Dialect/Standard/
Dcanonicalize.mlir104 // CHECK-NEXT: %[[DIM:.*]] = load %[[SHP]][%[[IDX]]]
107 // CHECK: return %[[DIM]] : index
/external/perfetto/infra/perfetto.dev/
Dbuild.js239 const DIM = '\u001b[2m';
243 const ts = `[${DIM}${ms}${RST}]`;
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor_test.cc615 const int DIM = 2; in TEST() local
617 Tensor ids(DT_INT64, TensorShape({N, DIM})); in TEST()
666 const int DIM = 2; in TEST() local
668 Tensor ids(DT_INT64, TensorShape({N, DIM})); in TEST()
/external/perfetto/ui/
Dbuild.js470 const DIM = '\u001b[2m';
474 const ts = `[${DIM}${ms}${RST}]`;
/external/llvm-project/mlir/test/Conversion/StandardToLLVM/
Dconvert-dynamic-memref-ops.mlir514 // CHECK: ^bb1([[DIM:%.*]]: !llvm.i64, [[CUR_STRIDE:%.*]]: !llvm.i64):
516 // CHECK: [[COND:%.*]] = llvm.icmp "sge" [[DIM]], [[C0_]] : !llvm.i64
520 // CHECK: [[SIZE_PTR:%.*]] = llvm.getelementptr [[SHAPE_IN_PTR]]{{\[}}[[DIM]]]
522 // CHECK: [[TARGET_SIZE_PTR:%.*]] = llvm.getelementptr [[SIZES_PTR]]{{\[}}[[DIM]]]
524 // CHECK: [[TARGET_STRIDE_PTR:%.*]] = llvm.getelementptr [[STRIDES_PTR]]{{\[}}[[DIM]]]
527 // CHECK: [[STRIDE_COND:%.*]] = llvm.sub [[DIM]], [[C1_]] : !llvm.i64

123